2012-05-10 3 views
5

Non riesco a tracciare matplotlib.finance.candlestick senza i fine settimana (spazi vuoti tra ogni 5 candelabri). Lo example from Matplotlib's website non esclude nemmeno i fine settimana e sembra che the way to exclude weekends on other plots non si applichi a CandleSticks.Come faccio a tracciare solo i giorni feriali usando il candlestick matplotlib di Python?

Qualcuno si è imbattuto in questo prima?

ps. come richiesto, ecco l'esempio:

#!/usr/bin/env python 
from pylab import * 
from matplotlib.dates import DateFormatter, WeekdayLocator, HourLocator, \ 
DayLocator, MONDAY 
from matplotlib.finance import quotes_historical_yahoo, candlestick,\ 
plot_day_summary, candlestick2 

# (Year, month, day) tuples suffice as args for quotes_historical_yahoo 
date1 = (2004, 2, 1) 
date2 = (2004, 4, 12) 


mondays = WeekdayLocator(MONDAY)  # major ticks on the mondays 
alldays = DayLocator()    # minor ticks on the days 
weekFormatter = DateFormatter('%b %d') # Eg, Jan 12 
dayFormatter = DateFormatter('%d')  # Eg, 12 

quotes = quotes_historical_yahoo('INTC', date1, date2) 

fig = figure() 
fig.subplots_adjust(bottom=0.2) 
ax = fig.add_subplot(111) 
ax.xaxis.set_major_locator(mondays) 
ax.xaxis.set_minor_locator(alldays) 
ax.xaxis.set_major_formatter(weekFormatter) 

#plot_day_summary(ax, quotes, ticksize=3) 
candlestick(ax, quotes, width=0.6) 

ax.xaxis_date() 
ax.autoscale_view() 
setp(gca().get_xticklabels(), rotation=45, horizontalalignment='right') 

show() 
+0

Potresti postare un esempio minimo? –

+0

@Zhenya, ecco un esempio: http://matplotlib.sourceforge.net/examples/pylab_examples/finance_demo.html?highlight=candlestick – Fred

+0

Possibile duplicato di [Intraday candlestick charts utilizzando MatPlotLib] (http://stackoverflow.com/questions/9673988/intraday-candela-grafici-con-matplotlib) – lanery

risposta

3

Dopo la linea di 'citazioni':

weekday_quotes = [tuple([i]+list(quote[1:])) for i,quote in enumerate(quotes)] 

poi

candlestick(ax, weekday_quotes, width=0.6) 

Ciò tracciare i dati senza gli spazi tra i giorni feriali, ora è è necessario modificare gli xtick alle date, preferibilmente il lunedì. Assumendo che il prima citazione è stato un lunedi:

import matplotlib.dates as mdates 

ax.set_xticks(range(0,len(weekday_quotes),5)) 
ax.set_xticklabels([mdates.num2date(quotes[index][0]).strftime('%b-%d') for index in ax.get_xticks()]) 

Questo è abbastanza grave, ma sembra ottenere il lavoro fatto - buona fortuna!