如何用Python实现相对强度指数（RSI）

相对强度指数（RSI）是一种技术分析指标，用于衡量价格变动的强度，通常应用于股票、期货等市场。下面是用Python实现RSI的步骤：

1. 导入所需的库。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

2. 定义函数计算RSI。

def calculate_rsi(data, window):
    delta = data.diff()
    gain = delta.where(delta > 0, 0)
    loss = -delta.where(delta < 0, 0)
    avg_gain = gain.rolling(window=window).mean()
    avg_loss = loss.rolling(window=window).mean()
    rs = avg_gain / avg_loss
    rsi = 100 - (100 / (1 + rs))
    return rsi

3. 加载数据并调用函数计算RSI。

data = pd.read_csv('data.csv', parse_dates=['date'], index_col='date')
rsi = calculate_rsi(data['close'], 14)

4. 绘制RSI曲线。

fig, ax = plt.subplots(figsize=(10, 6))
ax.plot(rsi.index, rsi.values, label='RSI')
ax.axhline(y=30, color='red', linestyle='--')
ax.axhline(y=70, color='red', linestyle='--')
ax.set_xlabel('Date')
ax.set_ylabel('RSI')
ax.legend()
plt.show()

这里假设数据文件为data.csv，其中包含日期和收盘价两列数据。在计算RSI时，一般采用14天的窗口。绘制RSI曲线时，常常将30和70作为超买和超卖的界限。