python：talib 计算 RSI

原创 2025-03-16 12:17:24 1.1w 阅读 AI写同款

belldeep

1820粉丝584篇原创

关注

Python量化实战：用TA-Lib计算RSI指标并识别超买超卖信号（附完整代码）本文详细介绍了如何使用Python和TA-Lib库计算RSI指标，并识别超买超卖信号。通过完整的代码示例，展示了从数据获取、指标计算到信号识别和策略构建的全过程，帮助量化交易开发者快速掌握RSI指标的应用技巧。阅读详情

AI摘要

本博客介绍了一个Python脚本，该脚本利用TALIB库计算股票的相对强弱指标(RSI)，并绘制了股票收盘价及RSI曲线图。通过输入股票代码，脚本将获取股票数据，并计算6天和14天的RSI值。

RSI:相对强弱指标

talib_rsi.py

# -*- coding: utf-8 -*-
import os, sys
import tushare as ts
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
import talib

if len(sys.argv) ==2:
    code = sys.argv[1]
else:
    print('usage: python talib_rsi.py stockcode ')
    sys.exit(1)

if len(code) !=6:
    print('stock code length: 6')
    sys.exit(2)

df = ts.get_k_data(code)
if df.empty ==True:
    print(" df is empty ")
    sys.exit(2)

df = df[ df['date'] > '2024-01-01']
if len(df) <15:
    print(" len(df) <15 ")
    sys.exit(2)

# DataFrame 重建索引
df = df.reset_index(drop=True)

# 求ma10日均线
df['ma10'] = df['close'].rolling(window=10).mean()
df.index = pd.to_datetime(df.date)
#dw = pd.DataFrame()
# RSI:相对强弱指标
df['rsi_6'] = talib.RSI(df.close, timeperiod=6)
df['rsi14'] = talib.RSI(df.close, timeperiod=14)
print("rsi_6={0:.1f} , rsi14={1:.1f}".format(df['rsi_6'][-1], df['rsi14'][-1]))
#print(df[ df['rsi_6'] > 80])
#print(d