问题:有什么简单的方法可以在 Python 中绘制一个可以旋转的 3d 散点图吗?

目前我正在使用 matplotlib 绘制 3d 散点图,虽然它完成了工作,但我似乎无法找到一种方法来旋转它以更好地查看我的数据。

这是一个例子:

import pylab as p
import mpl_toolkits.mplot3d.axes3d as p3

#data is an ndarray with the necessary data and colors is an ndarray with
#'b', 'g' and 'r' to paint each point according to its class

...

fig=p.figure()
ax = p3.Axes3D(fig)
ax.scatter(data[:,0], data[:,2], data[:,3], c=colors)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
fig.add_axes(ax)
p.show()

我想要一个可以让我在执行期间执行此操作的解决方案,但只要我可以旋转它并且它很短/很快我就可以了。

这是将 PCA 应用于 iris 数据集后生成的图的比较:

1\。难以捉摸

2\。 matplotlib

Mayavi 使数据可视化变得更容易,但 MatPlotLib 看起来更专业。 Matplotlib 也更轻。

解答

使用mayavi,您可以创建这样的情节

import enthought.mayavi.mlab as mylab
import numpy as np
x, y, z, value = np.random.random((4, 40))
mylab.points3d(x, y, z, value)
mylab.show()

GUI 允许通过单击和拖动进行旋转,并通过右键单击和拖动进行放大/缩小。