如何控制python散点图中每个点的颜色和不透明度？

Question

我正在寻找一个使用不透明度表示强度的4D数据集（X，Y，Z，强度）的图形。 我还希望颜色也取决于Z变量，以更好地显示深度。

到目前为止，这是相关的代码，对于Python我是一个新手：

.
.
.
x_list #list of x values as floats
y_list #list of y values as floats
z_list #list of z values as floats
i_list #list of intensity values as floats

.
.
.

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

Axes3D.scatter(ax, x_list, y_list, z_list)
.
.
.

那我该怎么做呢？

我认为颜色可能是z_list和颜色图（例如hsv）之间的线性关系，而不透明度也可能是线性的，即i_list / max（i_list）或沿这些线的东西。

Answer 1

我将执行以下操作：

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

# choose your colormap
cmap = plt.cm.jet

# get a Nx4 array of RGBA corresponding to zs
# cmap expects values between 0 and 1
z_list = np.array(z_list) # if z_list is type `list`
colors = cmap(z_list / z_list.max())

# set the alpha values according to i_list
# must satisfy 0 <= i <= 1
i_list = np.array(i_list)
colors[:,-1] = i_list / i_list.max()

# then plot
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(x_list, y_list, z_list, c=colors)
plt.show()

这是x_list = y_list = z_list = i_list 。 您可以在此处选择任何颜色图，也可以自己制作 ：