python：talib 计算 SAR 用 pro_api

最新推荐文章于 2023-03-13 22:40:36 发布

原创

最新推荐文章于 2023-03-13 22:40:36 发布 · 2k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #tushare #talib #sar

该代码示例展示了如何利用Tushare Pro接口获取股票数据，并结合Talib库计算SAR指标。通过matplotlib库绘制股票收盘价及SAR散点图，帮助投资者分析股票趋势。

SAR指标解释：baike.baidu.com/item/SAR指标/6329095?fr=aladdin

SAR指标画图 talib_sar2.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_sar2.py stockcode ')
    sys.exit(1)

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

if code < '600000':
    ts_code = code +'.SZ'
else:
    ts_code = code +'.SH'

# 初始化pro接口
pro = ts.pro_api('your token')

dh = pro.daily(ts_code=ts_code, start_date='20200101')
if dh.empty ==True:
    print(" df is empty ")
    sys.exit(2)

df = dh.sort_values(by='trade_date')
#df = df.reset_index(drop=True)
# 为了与原来的 tushare 保持一致性，修改列名
df.rename(columns={'trade_date':'date','vol':'volume'}, inplace=True)
if len(df) <10:
    print(" len(df) <10 ")
    sys.exit(2)

#df = df.resample('B').ffill()
print(df.head())
#