使用candlestick_ohlc显示csv

Question

我尝试用熊猫做第一步。

经过几个成功的步骤后，我坚持完成以下任务：用OHLC条显示数据。

我从Google财经下载了Apple股票的数据并将其存储到* .csv文件。

经过大量搜索后，我编写了以下代码：

  import pandas as pd
  import numpy as np
  import matplotlib.pyplot as plt
  import matplotlib.dates as mdates
  import datetime as dt
  from matplotlib.finance import candlestick_ohlc

  #read stored data

  #First two lines of csv:
  #Date,Open,High,Low,Close
  #2010-01-04,30.49,30.64,30.34,30.57

  data = pd.read_csv("AAPL.csv")

  #graph settings
  fig, ax = plt.subplots()
  ax.xaxis_date()
  ax.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m-%d"))
  plt.xlabel("Date")
  plt.ylabel("Price")
  plt.title("AAPL")

  #convert date to float format 
  data['Date2'] = data['Date'].map(lambda d: mdates.date2num(dt.datetime.strptime(d, "%Y-%m-%d")))

  candlestick_ohlc(ax, (data['Date2'], data['Open'], data['High'], data['Low'], data['Close']))
  plt.show()

但它显示空图。 这段代码有什么问题？

感谢。

Answer 1

您需要更改最后一行以每天组合元组。以下代码：

start = dt.datetime(2015, 7, 1)
data = pd.io.data.DataReader('AAPL', 'yahoo', start)
data = data.reset_index()
data['Date2'] = data['Date'].apply(lambda d: mdates.date2num(d.to_pydatetime()))
tuples = [tuple(x) for x in data[['Date2','Open','High','Low','Close']].values]

fig, ax = plt.subplots()
ax.xaxis_date()
ax.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m-%d"))
plt.xticks(rotation=45)
plt.xlabel("Date")
plt.ylabel("Price")
plt.title("AAPL")
candlestick_ohlc(ax, tuples, width=.6, colorup='g', alpha =.4);

制作以下情节：

你可以进一步修补。