python tushare 画股价走势图

pip install tushare

stock_price.py

# coding: utf-8
import os, sys
import datetime
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_price.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)

# 画股价走势图
fig,axes = plt.subplots(2,1)
d2[['close','ma5','ma10','ma20']].plot(ax=axes[0], grid=True, title=code)
# 画股票成交量图
d2[['volume']].plot(ax=axes[1], grid=True)
plt.show()

运行 python stock_price.py 600030

参考书: [ 利用Python进行数据分析 ] P247