通过tushare的股票数据绘制股票各曲线图——RSI指标曲线

import tushare as ts
import numpy as np
import pandas as pd
import datetime as dt
import matplotlib.pyplot as mp
import matplotlib.dates as md
#引入自己的api，把股票数据引入放到df里面
pro = ts.pro_api('ee5e027680ee512ab9783ae3e78d766ada5b959b441f4b2bba7db241')
df = pro.daily(ts_code='000001.SZ', start_date='20210401', end_date='20210601')

RsiData = {
        "trade_date":[],
        "up_value":[],
        "down_value":[]   
        }



#计算N=12的RSI值
for i in (range(0,df.trade_date.size - 12)):
    RsiData.setdefault("up_value",[]).append(0)
    RsiData.setdefault("down_value",[]).append(0)
    RsiData.setdefault("trade_date",[]).append(df.trade_date[i])
    #print(RsiData)
    for j in (range(i,i + 13)):
        if df.change[j] > 0:
            RsiData["up_value"][i] = RsiData["up_value"][i] + df.change[j]
        else:
            RsiData["down_value"][i] = RsiData["down_value"][i] + df.change[j]

MyRsiData = pd.DataFrame(RsiData)

#print(MyRsiData)

#创建图形实例，设置xy轴
mp.figure("RSI Line", figsize=(15, 5),facecolor="lightgray")
mp.title("RSI Line", fontsize=18)
mp.xlabel("Date", fontsize=15)
mp.ylabel("RSI", fontsize=15)

mp.tick_params(labelsize=5) #设置刻度标签的字体大小
mp.grid(linestyle=":")   #显示网格


mp.plot(MyRsiData.trade_date, MyRsiData.up_value/(MyRsiData.up_value - MyRsiData.down_value)*100, 
        color="b", linestyle="-",linewidth=2, label="RSI N=12", alpha=0.3)


#最终展示
mp.legend() #图例
mp.gcf().autofmt_xdate() #使X轴坐标斜着显示
mp.show()

结果展示：