import tushare as ts
import pandas as pd
import numpy as np
import datetime
from dateutil.relativedelta import relativedelta
import matplotlib.pyplot as plt
ts.set_token('you token')
pro=ts.pro_api()
today = datetime.datetime.today().strftime('%Y%m%d') #获取今天的年月日
lastday = datetime.datetime.today() - datetime.timedelta(days=1) #获取前一天数据
lastday = lastday.strftime('%Y%m%d')
last_year = datetime.datetime.today() - relativedelta(months=12) #获取前一年的日期
last_year = last_year.strftime('%Y%m%d') # 转换成STR
Lastweek = datetime.datetime.today() - datetime.timedelta(days=7) #获取前一周的日期
Lastweek = Lastweek.strftime('%Y%m%d')
df = pro.index_daily(ts_code='000001.SH',start_date = last_year, end_date=today)
df['date'] = df['trade_date'].apply(lambda x: datetime.datetime.strptime(x, '%Y%m%d')) #字符串转戳
df = df.sort_values(by = 'date') #将时间戳转为升序
plt.xticks(rotation = 45) #X轴的刻度显示角度45°
plt.plot(df['date'],df['close'],color = 'red') #设置线的颜色
plt.rcParams['figure.figsize'] = (14,8) # 设置窗口大小
plt.title('SZ 000001.SH') # 添加标题
plt.xlabel(last_year + '>>>' + today) # 添加X轴的标签
plt.ylabel("price") #添加Y轴的标签
plt.show()