移动3D绘图以避免边距裁剪

Question

我试图找出如何在下方获取3D matplotlib图片以在画布上绘制更高的图像，这样它就不会被剪裁。这是我用来创建绘图的代码。我无法找到附加包含Z高程的文本文件的方法（在下面的代码中引用），但它只是一个包含由0到1之间的值组成的表面的2D数组。

import os
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from mpl_toolkits.mplot3d import Axes3D

nrow=30
ncol=100

f = open(r'C:\temp\fracEvapCume_200.txt','r')
fracEvapTS = np.loadtxt(f)
f.close()

X, Y = np.meshgrid(ncol, nrow)
Y3d, X3d = np.mgrid[0:Y, 0:X]

fig = plt.figure()
ax = fig.gca(projection='3d')
ax.auto_scale_xyz([0, 100], [0, 30], [0, 0.2])
Y3d, X3d = np.mgrid[0:Y, 0:X]
Z = fracEvapTS

surf = ax.plot_surface(X3d, Y3d, Z, cmap='autumn', cstride=2, rstride=2)
ax.set_xlabel("X-Label")
ax.set_ylabel("Y-Label")
ax.set_zlabel("Z-Label")
ax.pbaspect = [1., .33, 0.25]
ax.dist  = 7
plt.tight_layout()
plt.savefig('clipped.png')

为了让ax.pbaspect=[1., .33, 0.25]行生效，我们按照this post中的建议更改了site-packages \ mpl_toolkits \ mplot3d \ axes3d.py中的get_proj函数。为了让数字更大，我根据this post添加了ax.dist = 7。最后，基于this post我希望plt.tight_layout()能够回滚边距并防止下面显示的红色/黄色表面被剪裁，但这也不起作用。我没有找到将图像向上移动到画布上的命令，从而避免了图形顶部的所有不必要的空白区域，并防止红色/黄色表面被剪裁。是否有一行Python可以实现这一目标？

添加行plt.tight_layout()后，情况更糟：

Answer 1

问题是你对 site-packages \ mpl_toolkits \ mplot3d \ axes3d.py 的修改会改变投影矩阵，而不会改变视图的中心，一旦变换就弄乱了场景的位置在相机坐标。

因此，当视图缩放（使用ax.dist）然后移动时，绘图有时会脱离画布。

您需要将以下行替换为 axes3d.py 中的get_proj函数：

    # look into the middle of the new coordinates
    R = np.array([0.5, 0.5, 0.5])

通过：

    # look into the middle of the new coordinates
    try:
        R = np.array(self.pbaspect)/2
    except AttributeError:
        R = np.array([0.5, 0.5, 0.5])

这应该有效：

PS：用于制作数字的代码：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from mpl_toolkits.mplot3d import Axes3D

nrow=30
ncol=100

X, Y = np.meshgrid(ncol, nrow)
Y3d, X3d = np.mgrid[0:Y, 0:X]
Z = np.sin(Y3d/Y)*np.sin(X3d/X)

fig = plt.figure()
for i in range(4):
    ax = fig.add_subplot(2,2,i,projection='3d')
    ax.auto_scale_xyz([0, 100], [0, 30], [0, 0.2])

    surf = ax.plot_surface(X3d, Y3d, Z, cmap='autumn', cstride=2, rstride=2)
    ax.set_xlabel("X-Label")
    ax.set_ylabel("Y-Label")
    ax.set_zlabel("Z-Label")
    ax.pbaspect = [1., .33, 0.25]
    ax.dist  = 7