CCI指标解释:baike.baidu.com/item/CCI顺势指标
talib_cci.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_cci.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'] > '2020-01-01']
if len(df) <10:
print(" len(df) <10 ")
sys.exit(2)
df['ma10'] = df['close'].rolling(window=10).mean()
df.index = pd.to_datetime(df.date)
# baike.baidu.com/item/CCI顺势指标
cci = talib.CCI(df.high, df.low, df.close, timeperiod=14)
print(cci[-5:])
# 画股票收盘价图
fig,axes = plt.subplots(2,1)
df[['close', 'ma10']].plot(ax=axes[0], grid=True, title=code)
# 画 cci 曲线图
cci.plot(ax=axes[1], grid=True, label='CCI')
plt.legend(loc='best', shadow=True)
plt.show()
运行 python talib_cci.py 000001