Python 3d scatterplot colormap问题

我有四维数据(x,y,z位移;和各自的电压),我希望在python中的3d散点图中绘制.我已经获得了渲染的3d图,但我希望使用颜色图来改变点的颜色,这取决于点电压的大小.

我已经尝试了一些东西,但似乎无法让它工作我收到错误ValueError：无法转换参数类型< type'numpy.ndarray'>到rgba数组.我不确定如何转换我需要转换的内容,所以如果有人可以提供一些帮助,我会非常感激.

我的代码在这里：

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()

解决方法:

ax.scatter可以采用颜色参数c,其是标量的序列(例如列表或数组),以及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()

标签：python,matplotlib,color-mapping
来源： https://codeday.me/bug/20190529/1179470.html