python tushare 画股票成交量图

pip install tushare

stock_vol.py

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

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

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

# help(ts.get_hist_data) 了解参数
dh = ts.get_hist_data(code)
df = dh.sort_values(by='date')
df.reset_index(inplace=True)
# 取样 2019年以后的数据
d2 = df[ df['date'] >'2019-01-01']
print(d2.head())

d2.index = pd.to_datetime(d2.date)
# 画股票成交量图
d2[['volume','v_ma5','v_ma10','v_ma20']].plot(kind='line',grid=True,title=code)
plt.show()

运行 python stock_vol.py 000001