zoukankan      html  css  js  c++  java
  • python 绘制走势图

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

  • 相关阅读:
    3、linux网络编程--网络字节序
    1、linux网络编程--网络协议
    第四篇:进程
    第五篇:进程通信
    第六篇:线程原理
    操作系统基本概念
    第一篇:基础原理篇
    第二篇:操作系统历史
    (2)linux下的简单的socket通信实例
    (3)基于linux的socket编程TCP半双工client-server聊天程序
  • 原文地址:https://www.cnblogs.com/colin2012/p/7770674.html
Copyright © 2011-2022 走看看