python 绘制走势图

Posted 2020-10-12

1.工具

  matplotlib

 numpy

2.matplotlib使用

import matplotlib.pyplot as plt #plt用于显示图片
import matplotlib.image as mping #mping用于读取图片
import datetime as dt
import matplotlib.dates as mdates
from pylab import *


def draw_trend_chart(dates,y):
    mpl.rcParams[‘font.sans-serif‘] = [‘SimHei‘] #指定默认字体
    mpl.rcParams[‘axes.unicode_minus‘] = False #解决保存图像是负号‘-‘显示为方块的问题
    
    x = [dt.datetime.strptime(d,‘%Y/%m/%d‘).date() for d in dates]
    #plt.figure(figsize=(8,8))
    plt.figure()

    #plt.gca().xaxis.set_major_formatter(mdates.DateFormatter(‘%m/%d/%Y‘))
    #plt.gca().xaxis.set_major_locator(mdates.DayLocator())
    #plt.plot(x,y,"r--",linewidth=2)
    plt.plot(x,y,"r",linewidth=1)
    #plt.gcf().autofmt_xdate()
    
    #plt.xlabel("DATE") #x轴标签
    plt.ylabel("WEIGHT") #y轴标签
    plt.title("MY HEALTH TRACKING")#标题

    plt.savefig("liuyang.png") #保存图片名称

    lena = mping.imread(‘liuyang.png‘)  #读取图片文件信息
    lena.shape #（512,512,3）

    plt.imshow(lena) #显示图片
    plt.axis(‘off‘) #不显示坐标轴
    plt.title("")
    plt.show() #显示

def get_weight_data(filename):
    time = []
    weight = []
    fileContent=open(filename,"r")
    for eachline in fileContent:
        eachData = eachline.strip(‘\\n‘).split(",")
        if eachData[-1].strip() ==‘‘:
            continue
        else:
            time.append(eachData[0])
            weight.append(eachData[1])
    return [time, weight]

data = get_weight_data("data.csv")
draw_trend_chart(data[0],data[1])

 

 

3.numpy

.numpy数组
numpy是常用的数据处理库，我将000001.SZ的股价数据（时间范围从20150101到20150930）导入到csv中，然后通过numpy读取得到日期和价格数组。csv文件中包含了两列，日期和价格，分割符号为","，读取过程代码如下：

dates, close = np.loadtxt(filename,delimiter=",", unpack= True, converters={0:mdates.strpdate2num(‘%Y-%m-%d‘)})

　　numpy的loadtxt方法：设定了文件中的分隔符号为“，”，unpack是否将数组拆分，True为拆分，即：得到两个数组分别表示日期和收盘价格，converters是将日期类型的字符串转为数组，因为numpy规定其数组中的类型需要一致。
2.绘制基础走势图
通过numpy得到日期和收盘价格之后，参照先前的示例，通过plot方法进行绘制：

ax1.plot(dates,close)

　　图表如下所示：

在上图中，x轴通过数值标识，而不是日期类型。因为我们通过numpy构造数组时，日期按照数值类型存储，可以通过如下方法转换为日期类型：

    ax1.xaxis.set_major_locator(mdates.DayLocator(bymonthday=range(1,32), interval=15)) 
    ax1.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m-%d"))
      for label in ax1.xaxis.get_ticklabels():   
       label.set_rotation(45)

　　X-轴设置主要刻度locator为每日刻度，格式为：DateFormatter("%Y-%m-%d")，每日刻度从第1日到第31日，间隔为15日。图表如下所示：

其中，对于X-轴上面的每个ticker标签都向右倾斜45度，这样标签不会重叠在一起便于查看。
也可以按照每月进行显示，X-轴设置主要刻度为每月刻度，格式为：DateFormatter("%Y-%m")，转换代码如下：

    ax1.xaxis.set_major_locator(mdates.MonthLocator()) 
     ax1.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m"))

　　图表如下所示：

3.图形边框调整
细心的读者可能会发现，在上图中底部（bottom）处的日期标签，显示不完整。此时可以点击【configure subplots】按钮，调整左右和上下边框，然后保存即可。图表如下所示：

也可以通过subplots_adjust方法对于边框进行调整：

    plt.subplots_adjust(bottom=0.13,top=0.95)

4.绘制价格平均线
在前面的示例中绘制了基本的股票走势图，本节我们利用ta-lib【4】证券技术指标库，绘制5日和10日价格平均线。
ta-lib中提供了方法talib.SMA得到价格简单平均线，timeperiod为时间参数，timeperiod=5为五日均线，基于上图增加五日和十日均线，代码如下：

    sma5 = talib.SMA(close, timeperiod = 5) 
     ax1.plot(dates,sma5)
      sma10 = talib.SMA(close, timeperiod = 10) 
     ax1.plot(dates,sma10)

　　其中，sma5和sma10均为numpy数组。
图表如下所示：

在坐标轴中有三条曲线，matplotlib会自动改变线条颜色，此时如果没有说明并不方便使用，可以在右上角增加图例，表明各线条所代表的含义，并增加图表的网格效果，代码如下：

plt.legend((‘daily‘, ‘SMA5‘, ‘SMA10‘)) 
 plt.grid(True)

　　图表如下所示：

 

 

 4.http://www.cnblogs.com/bradleon/p/6832867.html