from datetime import datetime, date, timedelta
import matplotlib.pyplot as plt
import tushare as ts
plt.rcParams['font.sans-serif'] = ['SimHei'] #显示中文
plt.rcParams['axes.unicode_minus']=False #用来正常显示负号
def get_riqi():
count = 30
i = 0
today = date.today() #今天日期
riqi_list=[]
riqi_list.append(str(today))
while(i<=count):
i+=1
riqi_list.append(str((today + timedelta(days=-i))))
return riqi_list
def count(Data):
data_list = []
sum_list = []
for data in Data:
sum = 0
list = []
for jiage in data:
sum+=jiage
list.append(jiage)
sum_list.append(round(sum,2))
data_list.append(list)
return sum_list,data_list
def get_juedui(Data):
sum = 0
data = []
for i in Data[::-1]:
sum += i
data.append(sum)
return data
technology = {
'海康威视':'002415',
'中兴通讯':'000063',
'科大讯飞':'002230',
'立讯精密':'002475',
'蓝思科技':'300433',
'歌尔股份':'002241'
}
store = [technology]
technology = []
def get_data(store,riqi_list):
count = 0
for Stock_pool in store:
count +=1
for stock in Stock_pool.values():
data = ts.get_hist_data(stock,start=riqi_list[-1])
if (count == 1):
technology.append(data.loc[:, 'p_change'])
riqi_list = get_riqi()
get_data(store,riqi_list)
riqi_list = technology[0].index
rili = []
for riqi in riqi_list:
rili.append(riqi[5:])
sum2,data2 = count(technology)
riqi_list = [datetime.strptime(d, '%m-%d').date() for d in rili]
count = 0
for i in data2:
count+=1
data = get_juedui(i)
if(count==1):
plt.plot(rili[::-1], data,label = '海康威视' )
if(count==2):
plt.plot(rili[::-1], data,label = '中兴通讯' )
if(count==3):
plt.plot(rili[::-1], data,label = '科大讯飞' )
if(count==4):
plt.plot(rili[::-1], data,label = '立讯精密' )
if (count == 5):
plt.plot(rili[::-1], data, label='蓝思科技')
if (count == 6):
plt.plot(rili[::-1], data, label='歌尔股份')
plt.legend()
plt.show()
