如何为4维RGB NumPy数组添加Alpha通道？

参考Matplotlib彩色体素相关教程编写了彩色立方体生成代码，能正常生成立方体但完全不透明。尝试用np.dstack添加Alpha通道却无效，现请教正确的Alpha通道添加方式。

我的核心代码如下：

spatialAxes = [self.step, self.step, self.step]
r, g, b= np.indices((self.step+1, self.step+1, self.step+1)) / 16.0
rc = self.midpoints(r)
gc = self.midpoints(g)
bc = self.midpoints(b)

cube = np.ones(spatialAxes, dtype=np.bool)

# 组合颜色分量
colors = np.zeros(cube.shape + (3,))
colors[..., 0] = rc
colors[..., 1] = gc
colors[..., 2] = bc

self.axes.voxels(r, g, b, cube,
      facecolors = colors, 
      linewidth = 0)

其中用到的midpoints函数定义：

def midpoints(self, x):
    sl = ()
    for i in range(x.ndim):
        x = (x[sl + np.index_exp[:-1]] + x[sl + np.index_exp[1:]]) / 2.0
        sl += np.index_exp[:]
    return x

我尝试添加透明度的代码：

colors = np.dstack( (colors, np.ones(spatialAxes) * .5) ) # 设置0.5透明度

但该方法无效，已知colors是形状为(X, X, X, 3)的4维数组（X为self.step的值）。

问题出在np.dstack的使用上——它会改变数组的维度排列顺序，导致和voxels函数期望的形状不匹配。正确的做法是直接扩展数组的最后一维来添加Alpha通道：

1. 正确构造RGBA数组 #

可以选择两种方式构造4通道的RGBA数组：

• 方式一：直接创建RGBA数组
  初始化时就设置4个通道，最后一个通道赋值为透明度：

colors = np.zeros(cube.shape + (4,))
colors[..., 0] = rc
colors[..., 1] = gc
colors[..., 2] = bc
colors[..., 3] = 0.5  # 透明度设为0.5，范围0-1

• 方式二：在原3通道数组后追加Alpha通道
  用np.concatenate在最后一维拼接Alpha通道数组：

# 创建和颜色数组匹配形状的Alpha通道数组
alpha = np.full(cube.shape + (1,), 0.5)
# 拼接成RGBA数组
colors = np.concatenate([colors, alpha], axis=-1)

2. 注意Matplotlib 3D体素的透明度限制 #

Matplotlib的3D体素透明度渲染有一些局限性：

• 体素的透明度是面级别的，内部重叠面的透明度可能无法正确显示（因为Matplotlib 3D渲染不处理复杂的深度排序）
• 确保使用较新版本的Matplotlib，旧版本对3D体素RGBA颜色的支持可能不完善

3. 验证数组形状 #

修改完成后，colors数组的形状应为(X, X, X, 4)，和cube的(X,X,X)形状对应，每个体素对应一个RGBA颜色值。

内容的提问来源于stack exchange，提问作者Amarth Gûl