如何缩放3D和2D子图,以使相应的轴具有相同的长度?
问题描述
我正在尝试使用两个子图来缩放图形(请参见下图),以便3D绘图的z轴与2D绘图的y轴的大小相同。我该怎么做?我已经尝试了GRIDSPEC和方面,但不起作用...
解决方案
确定3D轴的确切位置和长度并通过2D图形复制它是相当困难的(尽管并非不可能-我将支持任何完全做到这一点的解决方案)。但如果你做不到,那就假装吧。在这里,我们使用两个3D绘图,并将其中一个转换为看起来像2D绘图:
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d.axes3d import Axes3D
def plot_3d_pseudo2d(arr):
#one figure, two 3D subplots
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(20, 10), subplot_kw={'projection': "3d"})
#ax1 - normal isometric 3D projection
ax1.view_init(elev=10, azim=-45)
ax1.set_xlabel("X")
ax1.set_ylabel("Y")
ax1.set_zlabel("Z")
ax1.plot(arr[:, 0], arr[:, 1], arr[:, 2], marker="o")
#pseudo-2D projection
ax2.view_init(elev=0, azim=-90.1)
#shrink y-axis projection length
ax2.get_proj = lambda: np.dot(Axes3D.get_proj(ax2), np.diag([1, 0.01, 1, 1]))
ax2.set_xlabel("X")
ax2.set_zlabel("Z")
#plot x-z pairs for y=0
ax2.plot(arr[:, 0], np.zeros(arr[:, 1].size), arr[:, 2], marker="o")
#remove y-ticks
ax2.set_yticks([])
#increase label distance
ax2.xaxis.labelpad = 10
ax2.zaxis.labelpad = 10
plt.subplots_adjust(wspace=0.1)
plt.show()
import numpy as np
plot_3d_pseudo2d(np.asarray([[1, 4, 5],
[12, 23, 89],
[123, 234, 789]]).T)
示例输出:
这并不完美-3D投影在它们周围创建了许多空白(以便有旋转的空间),而透视扭曲隐藏了x轴的刻度线,并将z轴标签设置为略有关闭。
免责声明:y轴投影因help of this answer而缩小。
相关文章