matplotlib bar3d剪辑问题

时间:2013-09-03 22:43:29

标签: python matplotlib

我正在尝试使用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时遇到无效值])

我的问题:

  1. 这些严重的警告吗?我需要调查它们并尝试 消除它们?我该如何消除它们?
  2. zsort ='max'和zsort ='average'之间有什么区别?
  3. 我还能做些什么来消除渲染问题?
  4. 提前致谢!

    这是我的代码:

    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)
    

4 个答案:

答案 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顺序吧(看看前面的高个子跳到他的短小朋友面前):

wrong order of bars

条形图的错误排序似乎与零除错误相关,因为当我使用较少数量的条形图时,图形看起来很好。

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)

生成数据图: enter image description here

我希望这很快得到解决。我想在不久的将来制作一些类似的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()

产生下图: enter image description here

注意:这仅适用于初始视角。如果旋转轴,则必须再次为所有条设置_sort_zpos并重绘画布以修复渲染。

答案 2 :(得分:2)

我拿了apodemus's code,显然有效,将其解压缩,并将其应用于原始问题,以便为其提供直接答案。我的代码肯定可以清理 - 尤其是getDistances()中的循环 - 但它确实解决了所提出的问题,并且应该更容易理解。为了推测,必须通过调用sphview()sph2cart()来确定到观看者的距离,即相机距离。然后,必须通过调用getdistances()计算所有条形距相机的距离。此后,应该一次和一次绘制条形图,并且至关重要的是,必须根据先前确定的距离明确设置每个条形的z顺序。

如果生成的图形在绘图窗口中实时旋转,则可能无法正确更新。但是,预设摄像机的位置可以毫无错误地绘制任意初始视图。 (可能会有一个回调机制可以调用,以便明确重新计算条形的z顺序,但我不知道这样的API。)可以通过传递{{1来预设摄像机的位置。 }和azimelev。可以通过设置fig.add_subplot()返回的Axes实例的dist字段来更改其距离。

以下是应用于原始问题的更新代码生成的图表:

enter image description here

fig.add_subplot()

这种方法(例如,与使用Mayavi处理3D绘图相反)允许matplotlib外观保留在图形本身以及轴数,标签和图例等装饰中。

答案 3 :(得分:1)

我认为marisano的答案存在渲染各种高度的问题,因为它使用从条形顶部到摄像机位置的欧几里德距离,并从最大z_order中减去此值,I不要以为这是正确的方法。最后,我对apodemus中的z_order进行了相同的测量,并在下面进行了更新:

z_order = np.multiply([xpos,ypos, np.zeros_like(xpos)],camera).sum(0)

pl._sort_zpos = z_order[i]

现在它适用于我的情况。