python - Python 3d散点图颜色图问题

Question

我有四维数据（x、y、z 位移；以及各自的电压），我希望在 python 中绘制 3d 散点图。我已经得到了要渲染的 3d 图，但是我想使用颜色图来改变点的颜色，这取决于点电压的大小。

我已经尝试了一些东西，但似乎无法让它工作我得到了错误ValueError: Cannot convert argument type <type 'numpy.ndarray'> to rgba array。我不确定如何转换我需要转换的东西，所以如果有人可以提供一些帮助，我将不胜感激。

我的代码在这里：

fig = plt.figure()
from mpl_toolkits.mplot3d import Axes3D
cmhot = plt.cm.get_cmap("hot")
ax = fig.add_subplot(111, projection='3d',)
ax.scatter(x, y, z, v, s=50, c = cmhot)
plt.show()

Answer 1

ax.scatter可以采用一个颜色参数c，它是一个标量序列（例如 alist或 an array），以及一个cmap用于指定颜色图的参数。因此，要使颜色根据电压的大小而变化，您可以定义：

c = np.abs(v)

这使得正负电压具有相同的颜色。相反，如果您希望每种颜色（正或负）都有自己的颜色，则可以使用c = v.

---

例如，

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
x, y, z, v = (np.random.random((4,100))-0.5)*15
c = np.abs(v)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
cmhot = plt.get_cmap("hot")
cax = ax.scatter(x, y, z, v, s=50, c=c, cmap=cmhot)

plt.show()