python包使用tushare来获得股票数据，Tushare社区目前主要维护新版本：tushare pro，基本股票信息需要120积分即可获得（注册得100积分，补充个人信息得20积分）。下面展示如何使用tushare来获得某个股票，并将其日K线、5日均线、10日均线、20日均线可视化
import tushare as ts
import pandas as pd
import time
pro = ts.pro_api('your token')
df = pro.daily(ts_code='600754.SH',start_date='20200301',end_date='20200401') #随意选择一个股票，选择好需要分析的初始日期和结束日期
得到股票600754.SH的结果如下所示：
![]()
计算5日、10日、20日均线代码
df.sort_values(by='trade_date',inplace=True) # pro daily展示的结果是按照日期倒序，所以需要逆转过来计算五日、十日等均线
# 计算5日、10日、20日均线
ma_list = [5, 10, 20]
for ma in ma_list:
df['ma_' + str(ma)] = df['close'].rolling(ma).mean()
print(df)得到股票600754.SH的5日、10日、20均线结果如下所示：
![]()
可视化股票600754.SH的日K线、5日均线、10日均线、20日均线：
df = df.set_index('trade_date',drop=False) # 按照trade_date重建索引
import matplotlib.pyplot as plt
df[['close','ma_5','ma_10','ma_20']].plot(figsize=(12,6))
plt.title('600754.SH 的 20200301-20200401')
plt.xlabel('日期')
plt.show()
![]() |
|
转播
