python - 如何在 matplotlib 中为 3D 散点图上的数据点着色

Question

编辑:我设法弄清楚发生了什么事。散点图有一个参数“线宽”(lw=n)，用于确定散点图绘图点周围线条的粗细。因为我的绘图点尺寸为 1 (s=1)，所以线条宽度太粗，实际上覆盖了绘图点的颜色。将线宽设置为 0 (lw=0) 的厚度应该可以解决问题。

我想生成数据点的 3D 散点图，根据 y 坐标的值对它们进行着色，但我无法设法让这些点实际着色。

如果数据点的值较低，则颜色应更接近色谱的蓝色端。如果该值较高，则颜色应更接近光谱的红端。

我已经成功地以 2D 形式绘制了我想要的内容，但在以 3D 形式复制该过程时遇到了困难。当前代码仅将点绘制为黑色。

这是我的 3D 尝试代码，以及 2D 所需结果的屏幕截图。我到底做错了什么？

x_points、y_points 和 z_points 是浮点值列表。

import matplotlib.pyplot as plt
from matplotlib import cm
from mpl_toolkits.mplot3d import Axes3D

def three_dimensional_scatterplot(
    self, x_points, y_points, z_points, data_file
):

    cm1 = cm.get_cmap('gist_rainbow')

    fig = plt1.figure()
    ax = fig.add_subplot(111, projection='3d')
    ax.scatter(
        x_points,
        y_points,
        z_points,
        s=1,
        c=y_points,
        cmap=cm1
    )

    ax.set_xlabel('X axis')
    plt1.show()

Answer 1

你必须像这样绘制:

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

x = np.random.rand(25)
y = np.random.rand(25)
z = np.random.rand(25)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
p3d = ax.scatter(x, y, z, s=30, c=y, cmap = cm.coolwarm)
plt.show()