如何在Matplotlib中绘制带有颜色条的3D散点图？

要在Matplotlib中绘制带有颜色条的3D散点图，我们可以使用 scatter() 和 colorbar() 方法。

步骤

• 设置图形大小并调整子图之间和周围的填充。

• 创建一个新图或使用 figure() 方法激活现有图。

• 添加轴作为子图排列。

• 使用numpy创建xs、ys和zs数据点。

• 使用 scatter() 方法创建散点图。

• 使用 colorbar() 方法和 scatter scalar 可映射实例为颜色条。

• 要显示图形，请使用 show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
fig = plt.figure()
ax = fig.add_subplot(projection="3d")
xs = np.random.rand(100)
ys = np.random.rand(100)
zs = np.random.rand(100)
sc = ax.scatter(xs, ys, zs, c=xs, cmap="copper")
plt.colorbar(sc)
plt.show()

输出