Сделать прозрачную цветную полосу с высотой 0 в matplotlib

Question

У меня есть 3D-гистограмма, показывающая пропускную способность сети двух трех регионов. Я хочу сделать бары с 0 высотой прозрачными в matplotlib. В моем выводе они получают цвета, образуя квадрат с высотой 0.

Как я могу это сделать?

Код:

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

data = np.array([
[1000,200],
[100,1000],
])

column_names = ['','Oregon','', 'Ohio','']
row_names = ['','Oregon','', 'Ohio']

fig = plt.figure()
ax = Axes3D(fig)

lx= len(data[0])            # Work out matrix dimensions
ly= len(data[:,0])

xpos = np.arange(0,lx,1)    # Set up a mesh of positions
ypos = np.arange(0,ly,1)
xpos, ypos = np.meshgrid(xpos+0.25, ypos+0.25)

xpos = xpos.flatten()   # Convert positions to 1D array
ypos = ypos.flatten()
zpos = np.zeros(lx*ly)

dx = 0.5 * np.ones_like(zpos)
dy = dx.copy()
dz = data.flatten()

cs = ['r', 'g'] * ly

ax.bar3d(xpos,ypos,zpos, dx, dy, dz, color=cs)

ax.w_xaxis.set_ticklabels(column_names)
ax.w_yaxis.set_ticklabels(row_names)
ax.set_zlabel('Mb/s')

plt.show()

Answer 1

Существует параметр alpha, который можно установить для прозрачности, который немного скрыт в документации bar3d, поскольку он содержится в **kwargs, доступном из mpl_toolkits.mplot3d.art3d.Poly3DCollection.

Однако пока нет возможности дать массив значений alpha для каждого отдельного бара, как, например, для клавиши color. Следовательно, вы должны построить прозрачные и непрозрачные отдельно с помощью маски, например.

# needs to be an array for indexing
cs = np.array(['r', 'g', 'b'] * ly)

# Create mask: Find values of dz with value 0
mask = dz == 0

# Plot bars with dz == 0 with alpha
ax.bar3d(xpos[mask], ypos[mask], zpos[mask], dx[mask], dy[mask], dz[mask],
         color=cs[mask], alpha=0.2)
# Plot other bars without alpha
ax.bar3d(xpos[~mask], ypos[~mask], zpos[~mask], dx[~mask], dy[~mask], dz[~mask],
         color=cs[~mask])

Это дает