python matplotlib以日期为x轴作图

Posted 2020-11-22 -零

 

 

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()