tushare 写一段rsi计算代码?
时间: 2024-03-02 15:25:17 浏览: 24
以下是使用tushare计算RSI指标的Python代码:
```python
import tushare as ts
# 获取股票数据
df = ts.get_k_data('600519', start='2020-01-01', end='2020-12-31')
# 计算RSI指标
n = 14 # RSI的计算周期
up, down = [i for i in df['close']], [0] * len(df)
for i in range(1, len(df)):
if df['close'][i] > df['close'][i-1]:
up[i] = df['close'][i] - df['close'][i-1]
elif df['close'][i] < df['close'][i-1]:
down[i] = df['close'][i-1] - df['close'][i]
up_avg, down_avg = [0] * len(df), [0] * len(df)
up_avg[n-1] = sum(up[:n]) / n
down_avg[n-1] = sum(down[:n]) / n
for i in range(n, len(df)):
up_avg[i] = (up_avg[i-1] * (n-1) + up[i]) / n
down_avg[i] = (down_avg[i-1] * (n-1) + down[i]) / n
rs = [0] * len(df)
for i in range(n, len(df)):
rs[i] = up_avg[i] / down_avg[i]
rsi = [0] * len(df)
for i in range(n, len(df)):
rsi[i] = 100 - 100 / (1 + rs[i])
# 输出RSI指标
print(rsi)
```
这段代码使用tushare库获取股票数据,然后按照RSI指标的计算公式,自行编写了RSI指标的计算方法。最后输出计算出来的RSI指标数组。
相关推荐
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)