python 写一个rsi股票函数

以下是一个简单的Python函数，用于计算RSI（相对强弱指数）股票指标：

def calculate_rsi(prices, n=14):
    deltas = np.diff(prices)
    seed = deltas[:n+1]
    up = seed[seed >= 0].sum()/n
    down = -seed[seed < 0].sum()/n
    rs = up/down
    rsi = np.zeros_like(prices)
    rsi[:n] = 100. - 100./(1.+rs)

    for i in range(n, len(prices)):
        delta = deltas[i-1] 

        if delta > 0:
            upval = delta
            downval = 0.
        else:
            upval = 0.
            downval = -delta

        up = (up*(n-1) + upval)/n
        down = (down*(n-1) + downval)/n

        rs = up/down
        rsi[i] = 100. - 100./(1.+rs)

    return rsi

该函数使用Numpy库来处理价格数据，可以输入一个价格列表和一个可选的RSI计算周期（默认为14）。它首先计算价格差异，然后使用该差异值计算RSI值。由于RSI计算需要使用先前的值，因此该函数采用循环来计算RSI值并更新先前的值。最后，该函数返回一个RSI值列表，其长度等于输入价格列表的长度。