如何缩放3D和2D子图,以使相应的轴具有相同的长度?

2022-04-09 00:00:00 python matplotlib 3d 2d scale

问题描述

我正在尝试使用两个子图来缩放图形(请参见下图),以便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而缩小。

相关文章