三维散点图

Question

我想创建一个3D散点图。第四个参数将由颜色图表示。我可以用Python和IDL编程，但是想知道是否有人创建了这样的散点图以及使用的最佳工具是什么？

感谢。

Answer 1

有两个主要选项matplotlib和Mayavi。目前matplotlib更受欢迎，因此您可以更轻松地找到信息。 Mayavi往往更高级，因此如果matplotlib不足以满足您的使用需求（例如太慢），您可以使用它。使用matplotlib的一个例子：

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
n = 100

xs = np.random.randint(0, 10, size=n)
ys = np.random.randint(0, 10, size=n)
zs = np.random.randint(0, 10, size=n)
colors = np.random.randint(0, 10, size=n)
ax.scatter(xs, ys, zs, c=colors, marker='o')
plt.show()

您可以找到更多示例here。