如何消除 3D 投影图中的偏移？

Question

我意识到我用 matplotlib 制作的 3D 绘图有轻微的“错位”。这是 MWE：

import numpy as np
from matplotlib import pyplot as plt
figure = plt.figure(figsize=(8,10.7))
ax = plt.gca(projection='3d')
ax.plot_surface(np.array([[0, 0], [30, 30]]),
                np.array([[10, 10], [10, 10]]),
                np.array([[10, 20], [10, 20]]), 
                rstride=1, cstride=1
)
ax.plot_surface(np.array([[0, 0], [30, 30]]),
                np.array([[20, 20], [20, 20]]),
                np.array([[10, 20], [10, 20]]), 
                rstride=1, cstride=1
)
plt.show()
plt.close()

显然，箱没有正确居中，因为表面似乎从 10.5 开始并以 20.5 结束，而不是急剧的 10 和 20。如何才能实现后者呢？

编辑：恐怕建议的答案有问题。 x 轴没有黑色实线，默认情况下就是这样：

当我取出建议的包装时，我得到：

不幸的是，当我拿出我的东西时在绘图时，这个问题在 Jupyter 笔记本中无法重现，但尽管如此，我想知道您是否能够向我指出我必须做什么，以便在我的情况下，x 轴有一个黑色又线了？

Answer 1

这是由 matplotlib 对 3D Axis 坐标的处理引起的。它故意改变 mins 和 maxs 来创建一些人工填充：

<代码> axis3d.py#L190-L194

类轴（maxis.XAxis）：
   ...
   def _get_coord_info(自身，渲染器)：
       ...
       # 在最小/最大点和绘图边缘之间添加一个小的偏移量
       增量 =（最大值 - 分钟）/ 12
       分钟 -= 0.25 * 增量
       最大值 += 0.25 * 增量
       ...
       返回最小值、最大值、中心、增量、bounds_proj、最高值

从 v3.5.1 开始，没有参数可以控制此行为。

但是，我们可以使用 functools.wraps< /a> 创建一个围绕 Axis._get_coord_info 的包装器，用于取消移动 mins 和 maxs。为了防止此包装器多次恢复原位（例如，重新运行其 Jupyter 单元时），请通过 跟踪包装器状态 >_unpadded 属性：

from functools import wraps

def unpad(f): # where f will be Axis._get_coord_info
    @wraps(f)
    def wrapper(*args, **kwargs):
        mins, maxs, centers, deltas, bounds_proj, highs = f(*args, **kwargs)
        mins += 0.25 * deltas # undo original subtraction
        maxs -= 0.25 * deltas # undo original addition
        return mins, maxs, centers, deltas, bounds_proj, highs

    if getattr(f, '_unpadded', False): # bypass if already unpadded
        return f
    else:
        wrapper._unpadded = True # mark as unpadded
        return wrapper

在绘图之前应用 unpad 包装器：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.axis3d import Axis

X = np.array([[0, 0], [30, 30]])
Z = np.array([[10, 20], [10, 20]])
y1, y2, y3 = 10, 16, 20

# wrap Axis._get_coord_info with our unpadded version
Axis._get_coord_info = unpad(Axis._get_coord_info)

fig, ax = plt.subplots(figsize=(8, 10.7), subplot_kw={'projection': '3d'})
ax.plot_surface(X, np.tile(y1, (2, 2)), Z, rstride=1, cstride=1)
ax.plot_surface(X, np.tile(y2, (2, 2)), Z, rstride=1, cstride=1)
ax.plot_surface(X, np.tile(y3, (2, 2)), Z, rstride=1, cstride=1)

plt.show()