python matplotlib以日期为x轴作图

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

猜你喜欢

转载自www.cnblogs.com/-wenli/p/11908743.html
今日推荐