2012-05-10 49 views
5

我沒有設法在沒有周末的情況下繪製matplotlib.finance.candlestick(每5根蠟燭之間的空白)。 example from Matplotlib's website不排除週末,the way to exclude weekends on other plots似乎不適用於CandleSticks。如何僅使用Python的matplotlib燭臺繪製工作日?

有沒有人遇到過這個?

ps。根據要求,這裏是例子:

#!/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

請問您可以發表最簡單的例子嗎? –

+0

@ Zhenya,這裏是一個例子:http://matplotlib.sourceforge.net/examples/pylab_examples/finance_demo.html?highlight=candlestick – Fred

+0

可能重複[日內燭臺圖表使用MatPlotLib](http://stackoverflow.com/questions/9673988 /盤中-燭臺圖-使用-matplotlib) – lanery

回答

3

你的 '報價' 行後:

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

然後

candlestick(ax, weekday_quotes, width=0.6) 

這將繪製數據,而不工作日之間的差距,現在你必須將xticks改回日期,最好是星期一。假設你的第一個報價是星期一:

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()]) 

這很粗糙,但似乎完成了工作 - 祝你好運!