具有350个点的3D散点图上的颜色代码点

Question

我正在做PCA，它有350个点要绘制。我已经成功地将它们绘制在3D图上。我想给它们上色，告诉他们哪个点是我3D图上的哪个点。有没有可能给每一个都涂上颜色？也许颜色指向1到50种不同深浅的绿色，然后是51到100种不同深浅的红色，等等？颜色越浅，数字越小？

Answer 1

你提到过你一直在尝试用matplotlib和ax.scatter来做到这一点。我认为对于您的示例，您需要的所有功能都已经通过c输入参数内置到了ax.scater中。例如，以下内容是否符合您的要求？

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

# Some dummy data
xval = np.random.randn(350)
yval = np.random.randn(350)
zval = np.exp( - (xval**2 + yval**2)/0.5)

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

cax = ax.scatter(xval, yval, zval, cmap=plt.cm.viridis, c=zval) 
fig.colorbar(cax)

​

​