回答问题

我正在尝试创建一个简单的 3D 散点图,但我还想在同一个图上显示此数据的 2D 投影。这将允许显示在 3D 图中可能难以看到的这 3 个变量中的两个变量之间的相关性。

我记得以前在某个地方看到过这个,但无法再次找到它。

这是一些玩具示例:

x= np.random.random(100)
y= np.random.random(100)
z= sin(x**2+y**2)

fig= figure()
ax= fig.add_subplot(111, projection= '3d')
ax.scatter(x,y,z)

Answers

您可以使用plot方法并指定zdir来添加 3D 散点数据的 2D 投影:

import numpy as np
import matplotlib.pyplot as plt

x= np.random.random(100)
y= np.random.random(100)
z= np.sin(3*x**2+y**2)

fig= plt.figure()
ax= fig.add_subplot(111, projection= '3d')
ax.scatter(x,y,z)

ax.plot(x, z, 'r+', zdir='y', zs=1.5)
ax.plot(y, z, 'g+', zdir='x', zs=-0.5)
ax.plot(x, y, 'k+', zdir='z', zs=-1.5)

ax.set_xlim([-0.5, 1.5])
ax.set_ylim([-0.5, 1.5])
ax.set_zlim([-1.5, 1.5])

plt.show()