我正在尝试使用Matplotlib 1.2.0和Python 2.7.3创建3D条形图。我按照http://www.mail-archive.com/matplotlib-users@lists.sourceforge.net/msg19740.html中的建议逐一绘制了条形图,但我仍然遇到渲染问题(即条形图彼此叠加)。
此外,当我调用我的代码时,我得到以下内容:
/usr/apps/python/lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.py:1476:RuntimeWarning:在法线中除以n时遇到的除以零])
/usr/apps/python/lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.py:1476:RuntimeWarning:在法线中除以n时遇到无效值])
我的问题:
提前致谢!
这是我的代码:
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.colors as colors
import matplotlib.cm as cmx
# my data
dat = [2.31778665482167e-310, 0.006232785101850947, 0.0285075971030949, 0.0010248181570355695, 0.0048776795767614825, 0.02877090365176044, 0.002459331469834533, 0.0008594610645495889, 0.002919824084878003, 0.000968081117692596, 0.0, 0.0, 0.0319623949119874, 0.00568752311279771, 0.009994801469036968, 0.03248018520506219, 0.006686905726805326, 0.005987863156039365, 0.0072955095915350045, 0.005568911905473998, 0.0, 0.0, 0.0, 0.028483143996551524, 0.031030793902192794, 0.06125216053962635, 0.02935971973938871, 0.028507530280092265, 0.030112963748812088, 0.028293406731749605, 0.0, 0.0, 0.0, 0.0, 0.004510645022825792, 0.028998119822468988, 0.0013993630391143715, 0.0010726572949244424, 0.002288215944285159, 0.0006513973584945584, 0.0, 1.1625e-320, 1.15348834e-316, 2.3177866547513e-310, 0.0, 0.03148966953869102, 0.005215047563268979, 0.004491716298086729, 0.006010166308872446, 0.005186976949223524, 0.0, 0.0, 0.0, 0.0, 0.0, 1.107e-320, 0.02983657915729719, 0.028893006725328373, 0.030526067389954753, 0.028629390713739978, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0015217840289869456, 0.002751587509779179, 0.001413669523724954, 1.15348834e-316, 2.3177866547513e-310, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0024680339073824705, 0.0008254364860386303, 0.0, 0.0, 0.0, 9.965e-321, 1.15348834e-316, 2.3177866547513e-310, 0.0, 0.0, 0.0, 0.002621588539481613, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 9.41e-321, 1.15348834e-316, 2.3177866547513e-310]
dat = np.reshape(dat,[10,10],order='F')
lx = len(dat[0])
ly = len(dat[:,0])
n = lx*ly
# generate colors
cm = plt.get_cmap('jet')
vv = range(len(dat))
cNorm = colors.Normalize(vmin=0, vmax=vv[-1])
scalarMap = cmx.ScalarMappable(norm=cNorm, cmap=cm)
colorVals = [scalarMap.to_rgba(i) for i in range(ly)]
# generate plot data
xpos = np.arange(0,lx,1)
ypos = np.arange(0,ly,1)
xpos, ypos = np.meshgrid(xpos+0.25, ypos+0.25)
xpos = xpos.flatten()
ypos = ypos.flatten()
zpos = np.zeros(n)
dx = 0.5*np.ones_like(zpos)
dy = dx.copy()
dz = dat.flatten()
cc = np.tile(range(lx), (ly,1))
cc = cc.T.flatten()
# generate plot
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
opacity = 1
for i in range(n):
ax.bar3d(xpos[i], ypos[i], zpos[i], dx[i], dy[i], dz[i],
color=colorVals[cc[i]], alpha=opacity, zsort='max')
plt.autoscale(enable=True, axis='both', tight=True)
plt.grid()
plt.show(block=False)
答案 0 :(得分:14)
这不是您正在寻找的答案,但我认为这可能是matplotlib中的一个错误。我认为同样的问题是encountered here。根据{{3}},问题被描述为“难以处理”。
但对我而言似乎并不棘手。您只需要确定哪个对象更靠近查看器并相应地设置z顺序。所以,我认为问题可能只是一个错误。
如果我使用matplotlib 3D mplot3d FAQ并将“bins = 4”更改为“bins = 6”或更高的数字,那么我得到相同的“axes3d.py:1476:RuntimeWarning:遇到无效值in divide / for n in normals])“。此外,我可以重现错误的z顺序吧(看看前面的高个子跳到他的短小朋友面前):
条形图的错误排序似乎与零除错误相关,因为当我使用较少数量的条形图时,图形看起来很好。
axes.py中的第1476行是:
shade = np.array([np.dot(n / proj3d.mod(n), [-1, -1, 0.5]) for n in normals])
基本上,我认为它试图使用每个面的法向量来计算阴影。但是,一个或多个法向量是零,这不应该是这种情况。所以,我认为这只是matplotlib中的一些错误,可能由具有比我自己更多编程技能的人修复。
mplot3d常见问题是正确的,如果你想要一个更好的3D引擎,可以使用MayaVI。我用了
from mayavi import mlab
mlab.barchart(xpos,ypos,dz*100)
生成数据图:
我希望这很快得到解决。我想在不久的将来制作一些类似的3D条形图。
答案 1 :(得分:3)
这个答案是一个快速修复解决方案,允许您使用正确的渲染在matplotlib中生成某些类型的3D条形图。诀窍是A)单独绘制条形图,B)破解zsort算法强制对条形图进行排序w.r.t. "距离"从相机。这可以通过覆盖_sort_zpos
返回的PolyCollection3D实例的ax.bar3d
属性来完成。以下代码演示了使用从2D高斯绘制的数据的解决方案。
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from scipy.stats import multivariate_normal
def sph2cart(r, theta, phi):
'''spherical to cartesian transformation.'''
x = r * np.sin(theta) * np.cos(phi)
y = r * np.sin(theta) * np.sin(phi)
z = r * np.cos(theta)
return x, y, z
def sphview(ax):
'''returns the camera position for 3D axes in spherical coordinates'''
r = np.square(np.max([ax.get_xlim(), ax.get_ylim()], 1)).sum()
theta, phi = np.radians((90-ax.elev, ax.azim))
return r, theta, phi
def ravzip(*itr):
'''flatten and zip arrays'''
return zip(*map(np.ravel, itr))
#Generate data
res = 15
sl = slice(-3, 3, complex(res))
Y, X = np.mgrid[sl, sl]
grid = np.array([X, Y])
(dx,), (dy,) = 0.8*np.diff(X[0,:2]), 0.8*np.diff(Y[:2,0])
#2D Gaussian
mu = (0, 0)
covm = np.array([[ 0.8, 0.3],
[ 0.3, 0.5]])
rv = multivariate_normal(mu, covm)
Zg = rv.pdf(grid.transpose(1,2,0)).T
#generate the figure
fig, (ax1, ax2) = plt.subplots(1,2, subplot_kw=dict(projection='3d'))
#standard bar3d
ax1.set_title('Standard')
ax1.bar3d(X.ravel(), Y.ravel(), np.zeros(X.size), dx, dy, Zg.ravel(), '0.85')
#Fixed bar3d
ax2.set_title('Fixed')
xyz = np.array(sph2cart(*sphview(ax2)), ndmin=3).T #camera position in xyz
zo = np.multiply([X, Y, np.zeros_like(Zg)], xyz).sum(0) #"distance" of bars from camera
bars = np.empty(X.shape, dtype=object)
for i, (x,y,dz,o) in enumerate(ravzip(X, Y, Zg, zo)):
j, k = divmod(i, res)
bars[j, k] = pl = ax2.bar3d(x, y, 0, dx, dy, dz, '0.85')
pl._sort_zpos = o
plt.show()
注意:这仅适用于初始视角。如果旋转轴,则必须再次为所有条设置_sort_zpos并重绘画布以修复渲染。
答案 2 :(得分:2)
我拿了apodemus's code,显然有效,将其解压缩,并将其应用于原始问题,以便为其提供直接答案。我的代码肯定可以清理 - 尤其是getDistances()
中的循环 - 但它确实解决了所提出的问题,并且应该更容易理解。为了推测,必须通过调用sphview()
和sph2cart()
来确定到观看者的距离,即相机距离。然后,必须通过调用getdistances()
计算所有条形距相机的距离。此后,应该一次和一次绘制条形图,并且至关重要的是,必须根据先前确定的距离明确设置每个条形的z顺序。
如果生成的图形在绘图窗口中实时旋转,则可能无法正确更新。但是,预设摄像机的位置可以毫无错误地绘制任意初始视图。 (可能会有一个回调机制可以调用,以便明确重新计算条形的z顺序,但我不知道这样的API。)可以通过传递{{1来预设摄像机的位置。 }和azim
到elev
。可以通过设置fig.add_subplot()
返回的Axes实例的dist
字段来更改其距离。
以下是应用于原始问题的更新代码生成的图表:
fig.add_subplot()
这种方法(例如,与使用Mayavi处理3D绘图相反)允许matplotlib外观保留在图形本身以及轴数,标签和图例等装饰中。
答案 3 :(得分:1)