九、时间戳的转换

本教程的重点是将来自 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