• 九、时间戳的转换

    九、时间戳的转换

    本教程的重点是将来自 Yahoo finance API 的日期转换为 Matplotlib 可理解的日期。 为了实现它，我们要写一个新的函数，bytespdate2num。

    1. def bytespdate2num(fmt, encoding='utf-8'):
    2.     strconverter = mdates.strpdate2num(fmt)
    3.     def bytesconverter(b):
    4.         s = b.decode(encoding)
    5.         return strconverter(s)
    6.     return bytesconverter

    此函数接受数据，基于编码来解码数据，然后返回它。

    将此应用于我们的程序的其余部分：

    1. import matplotlib.pyplot as plt
    2. import numpy as np
    3. import urllib
    4. import matplotlib.dates as mdates
    6. def bytespdate2num(fmt, encoding='utf-8'):
    7.     strconverter = mdates.strpdate2num(fmt)
    8.     def bytesconverter(b):
    9.         s = b.decode(encoding)
    10.         return strconverter(s)
    11.     return bytesconverter
    14. def graph_data(stock):
    16.     stock_price_url = 'http://chartapi.finance.yahoo.com/instrument/1.0/'+stock+'/chartdata;type=quote;range=10y/csv'
    17.     source_code = urllib.request.urlopen(stock_price_url).read().decode()
    18.     stock_data = []
    19.     split_source = source_code.split('\n')
    20.     for line in split_source:
    21.         split_line = line.split(',')
    22.         if len(split_line) == 6:
    23.             if 'values' not in line and 'labels' not in line:
    24.                 stock_data.append(line)
    26.     date, closep, highp, lowp, openp, volume = np.loadtxt(stock_data,
    27.                                                           delimiter=',',
    28.                                                           unpack=True,
    29.                                                           # %Y = full year. 2015
    30.                                                           # %y = partial year 15
    31.                                                           # %m = number month
    32.                                                           # %d = number day
    33.                                                           # %H = hours
    34.                                                           # %M = minutes
    35.                                                           # %S = seconds
    36.                                                           # 12-06-2014
    37.                                                           # %m-%d-%Y
    38.                                                           converters={0: bytespdate2num('%Y%m%d')})
    40.     plt.plot_date(date, closep,'-', label='Price')
    42.     plt.xlabel('Date')
    43.     plt.ylabel('Price')
    44.     plt.title('Interesting Graph\nCheck it out')
    45.     plt.legend()
    46.     plt.show()
    49. graph_data('TSLA')

    如果你绘制 TSLA，结果图应该看起来像这样：

    九、时间戳的转换 - 图1