python-3.x - ValueError : Invalid RGBA argument. 为什么会这样？我该如何修复它？

Question

我正在尝试按照以下代码创建一个简单的多维数据集:

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


# Create axis
axes = [5,5,5]

# Create Data
data = np.ones(axes, dtype=np.bool)

# Controll Tranperency
alpha = 0.9

# Control colour RGBA colour
colors = np.empty(axes + [4], dtype=np.float32)

colors[0] = [1, 0, 0, alpha] # red
colors[1] = [0, 1, 0, alpha] # green
colors[2] = [0, 0, 1, alpha] # blue
colors[3] = [1, 1, 0, alpha] # yellow
colors[4] = [1, 1, 1, alpha] # grey

# Plot figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Voxels are used for customizations of sizes, positions, and colors.
ax.voxels(data, facecolors=colors, edgecolors='grey')
plt.show()

效果很好。但是当我更改axes = [10, 10, 10] 时，代码如下:

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


# Create axis
axes = [10, 10, 10]

# Create Data
data = np.ones(axes, dtype=np.bool)

# Controll Tranperency
alpha = 0.9

# Control colour RGBA colour
colors = np.empty(axes + [4], dtype=np.float32)

colors[0] = [1, 0, 0, alpha] # red
colors[1] = [0, 1, 0, alpha] # green
colors[2] = [0, 0, 1, alpha] # blue
colors[3] = [1, 1, 0, alpha] # yellow
colors[4] = [1, 1, 1, alpha] # grey

# Plot figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Voxels are used for customizations of sizes, positions, and colors.
ax.voxels(data, facecolors=colors, edgecolors='grey')
plt.show()

有时有效，有时无效，并抛出错误:ValueError: Invalid RGBA argument: 4.435719e+27。当我删除 data = np.ones(axes, type=np.bool) 中的 dtype 时，出现同样的错误。现在我无法调试无效的 RGBA 参数，因为我不明白是什么导致了错误。我读过this ，但似乎是关于无效形状的错误，而不是无效值的错误。

为什么会发生这个错误？我该如何修复它？非常感谢。

Answer 1

您收到此错误是因为 np.empty 创建了基本上随机填充的数组(有时使用空内存空间，这就是为什么它有时会为您工作)。对于 axes = [5, 5, 5] 来说这不是问题，因为您在指定颜色时填写了正确的 RGBA 值，但对于较大的轴，则效果不佳。

查看当轴为 [5, 5, 5] 时打印 colors 的结果与使用 [ 时它不适合您的结果10, 10, 10]

修复方法:使用 np.zeros 而不是 np.empty 来确保零是缺失值的结果:

axes = [10, 10, 10]
colors = np.zeros(axes + [4], dtype=np.float32)