Matplotlib:在3D条形图中在x轴上格式化日期

问题描述:

鉴于此3D bar graph sample code,如何将x轴上的数字数据转换为格式化的日期/时间字符串?我试图使用ax.xaxis_date()函数而没有成功。我也尝试使用plot_date(),它似乎不适用于3D条形图。下面是示例代码的修改版本来说明什么,我试图做的:Matplotlib:在3D条形图中在x轴上格式化日期

from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
import numpy as np 
import matplotlib.dates as dates 

dates = [dates.date2num(datetime.datetime(2009,3,12)), 
     dates.date2num(datetime.datetime(2009,6,9)), 
     dates.date2num(datetime.datetime(2010,1,1)), 
     #etc... 
     ] 

fig = plt.figure() 
ax = Axes3D(fig) 
for c, z in zip(['r', 'g', 'b', 'y'], [30, 20, 10, 0]): 
    xs = np.array(dates) 
    ys = np.random.rand(20) 
    ax.bar(xs, ys, zs=z, zdir='y', color=c, alpha=0.8) 

ax.set_xlabel('Date & Time') 
ax.set_ylabel('Series') 
ax.set_zlabel('Amount') 

plt.show() 

alt text http://matplotlib.sourceforge.net/_images/bars3d_demo1.png

+0

我只是想你的代码,我绘制的相当好(x值与之相匹配的日期)在我的设置。我使用Python 2.6.4和最新的matplotlib(版本0.99.1)。你需要更新你的Python包吗? – 2010-02-07 02:51:00

+0

你只是想让xticks成为字符串而不是数字吗? – Mark 2010-02-07 03:00:43

+0

是,或者是一个选项来传递一个函数名来动态格式化ticks,而不是事先生成所有的字符串。 – pokstad 2010-02-07 07:03:48

可能有一些混乱这里,Axes3D具有的属性w_xaxis,w_yaxis和w_zaxis的轴系而不是通常的xaxix,yaxis等。

UPDATE现在使用函数来格式化标签。

from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
import numpy as np 
import matplotlib.dates as dates 
import datetime, random 
import matplotlib.ticker as ticker 

def random_date(): 
     date = datetime.date(2008, 12,01) 
     while 1: 
     date += datetime.timedelta(days=30) 
     yield (date) 

def format_date(x, pos=None): 
    return dates.num2date(x).strftime('%Y-%m-%d') #use FuncFormatter to format dates 

r_d = random_date() 
some_dates = [dates.date2num(r_d.next()) for i in range(0,20)] 

fig = plt.figure() 
ax = Axes3D(fig,rect=[0,0.1,1,1]) #make room for date labels 

for c, z in zip(['r', 'g', 'b', 'y'], [30, 20, 10, 0]): 
    xs = np.array(some_dates) 
    ys = np.random.rand(20) 
    ax.bar(xs, ys, zs=z, zdir='y', color=c, alpha=0.8,width=8) 

ax.w_xaxis.set_major_locator(ticker.FixedLocator(some_dates)) # I want all the dates on my xaxis 
ax.w_xaxis.set_major_formatter(ticker.FuncFormatter(format_date)) 
for tl in ax.w_xaxis.get_ticklabels(): # re-create what autofmt_xdate but with w_xaxis 
     tl.set_ha('right') 
     tl.set_rotation(30)  

ax.set_ylabel('Series') 
ax.set_zlabel('Amount') 

plt.show() 

产地:

alt text http://www.imagechicken.com/uploads/1265570252049424000.png

+0

+1让我的工作更轻松的完美答案,谢谢! – pokstad 2010-02-08 01:44:12

+0

+1,我拼命寻找一种方法来定制mplot3d中的刻度标签。 – fotNelton 2010-04-30 12:36:06