matplotlib 如何使用变量alpha创建三维条形图

rryofs0p  于 2023-05-18  发布在  其他
关注(0)|答案(1)|浏览(208)

我使用matplotlib bar3d和RdBu colormap,希望在条之间具有可变的透明度(因此较小的条可以比较高的条更透明)。
下面是制作3d条形图的代码。数据存储在4x4矩阵“rho”中。目前,alpha保持在0.95,但如果能够控制每个条形图的alpha值,那将是非常好的。

xpos = np.arange(0,4,1)
ypos = np.arange(0,4,1)
xpos, ypos = np.meshgrid(xpos, ypos)
xpos = xpos.flatten()
ypos = ypos.flatten()
zpos = np.zeros(4*4)
dx = 0.5 * np.ones_like(zpos)
dy = dx.copy()
dz = rho.flatten()
nrm=mpl.colors.Normalize(-1,1)
colors=cm.RdBu(nrm(-dz))
alpha = 0.95
ax.bar3d(xpos,ypos,zpos, dx, dy, dz, alpha=alpha, color=colors, linewidth=0)
jaxagkaj

jaxagkaj1#

python 3.11.2matplotlib 3.7.1中测试

import matplotlib.pyplot as plt
import numpy as np
import matplotlib as mpl

xpos = np.arange(0,4,1)
ypos = np.arange(0,4,1)
xpos, ypos = np.meshgrid(xpos, ypos)
xpos = xpos.flatten()
ypos = ypos.flatten()
zpos = np.zeros(4*4)
rho = np.random.random((4,4))
dx = 0.5 * np.ones_like(zpos)
dy = dx.copy()
dz = rho.flatten()
nrm = mpl.colors.Normalize(-1,1)
colors = plt.cm.RdBu(nrm(-dz))
alpha = np.linspace(0.2, 0.95, len(xpos), endpoint=True)

fig = plt.figure()
ax = fig.add_subplot(projection='3d')

for i in range(len(xpos)):
    ax.bar3d(xpos[i],ypos[i],zpos[i], dx[i], dy[i], dz[i], alpha=alpha[i], color=colors[i], linewidth=0)
    
plt.show()

相关问题