График Matplotlib Candlestick выглядит странно

Я пытался построить график свечей, используя биржевые данные, извлеченные из quandl. Однако график выглядит жутковато, появилось всего несколько столбцов. Я понятия не имею, что происходит. Любая помощь приветствуется.

Вот код:

start=dt.datetime(2000,1,1)
end=dt.datetime(2017,9,20)
df=quandl.get('WIKI/TWTR',start_date=start,end_date=end)
date_val=[x for x in range(len(df.index))]
open_val=np.array(df['Adj. Open'],dtype=np.float64)
high_val=np.array(df['Adj. High'],dtype=np.float64)
low_val=np.array(df['Adj. Low'],dtype=np.float64)
close_val=np.array(df['Adj. Close'],dtype=np.float64)
ohlc_data=[date_val,open_val,high_val,low_val,close_val]
ax1=plt.subplot(111)
candlestick_ohlc(ax1,ohlc_data,width=0.9,colorup='g',colordown='r',alpha=0.8)
plt.show()

Это график:
Это график


person Daniel Xin Li    schedule 23.09.2017    source источник


Ответы (1)


Здесь есть две основные проблемы.

  • Даты должны быть числовыми значениями, соответствующими рассматриваемым датам.

    date_val=matplotlib.dates.date2num(df.index.to_pydatetime()) 
    
  • Второй аргумент candlestick_ohlc должен быть последовательностью последовательностей, то есть каждое данное должно быть в нем индивидуально как (time, open, high, low, close, ...).

Возможное решение:

import matplotlib.pyplot as plt
import matplotlib.dates
from matplotlib.finance import candlestick_ohlc
import datetime as dt
import quandl

start=dt.datetime(2000,1,1)
end=dt.datetime(2017,9,20)
df=quandl.get('WIKI/TWTR',start_date=start,end_date=end)

#convert dates to datetime, then to float
date_val=matplotlib.dates.date2num(df.index.to_pydatetime()) 

open_val=df['Adj. Open'].values
high_val=df['Adj. High'].values
low_val=df['Adj. Low'].values
close_val=df['Adj. Close'].values
# ohlc_data needs to be a sequence of sequences
ohlc_data=zip(*[date_val,open_val,high_val,low_val,close_val])

ax1=plt.subplot(111)
candlestick_ohlc(ax1,ohlc_data,colorup='g',colordown='r',alpha=0.8)

# Format x axis for dates
ax1.xaxis.set_major_formatter(matplotlib.dates.DateFormatter('%Y-%m-%d'))
ax1.figure.autofmt_xdate()
plt.show()

Или, более конкретно:

import matplotlib.pyplot as plt
import matplotlib.dates
from matplotlib.finance import candlestick_ohlc
import datetime as dt
import quandl

start=dt.datetime(2000,1,1)
end=dt.datetime(2017,9,20)
df=quandl.get('WIKI/TWTR',start_date=start,end_date=end)

#convert dates to datetime, then to float
df["Date"]=matplotlib.dates.date2num(df.index.to_pydatetime())
ohlc_data= df[["Date",'Adj. Open','Adj. High','Adj. Low','Adj. Close']].values

ax1=plt.subplot(111)
candlestick_ohlc(ax1,ohlc_data,colorup='g',colordown='r',alpha=0.8)

# Format x axis for dates
ax1.xaxis.set_major_formatter(matplotlib.dates.DateFormatter('%Y-%m-%d'))
ax1.figure.autofmt_xdate()
plt.show()

введите здесь описание изображения

person ImportanceOfBeingErnest    schedule 23.09.2017
comment
Это очень полезно. Большое спасибо! - person Daniel Xin Li; 24.09.2017