如何缩放3D和2D子图,以使相应的轴具有相同的长度? [英] How to scale 3D and 2D subplots so that a corresponding axis has the same length?

查看:22
本文介绍了如何缩放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而缩小。

这篇关于如何缩放3D和2D子图,以使相应的轴具有相同的长度?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆