Question

在使用gridspec直接指定绘图位置或者subplot2grid时，我在访问Matplotlib图中的现有子图时遇到问题。常规子图规范，例如add_subplot（211），返回现有轴（如果有）。使用gridspec / subplot2grid似乎会破坏任何现有的轴。如何使用gridspec / subplot2grid检索现有轴对象？这是预期的行为还是我在这里遗漏了什么？我想要一个解决方案，我不必为轴对象定义自己的占位符。

示例：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

x = np.linspace(0,10,100)
y1 = np.cos(x)
y2 = np.sin(x)

fig = plt.figure()
ax = fig.add_subplot(211)
ax.plot(x,y1, '-b')
ax = fig.add_subplot(212)
ax.plot(x,y2, '-b')
ax = fig.add_subplot(211) #here, the existing axes object is retrieved
ax.plot(x,y2, '-r')

fig = plt.figure()
gs = gridspec.GridSpec(2,1)
ax = fig.add_subplot(gs[0,0])
ax.plot(x,y1, '-b')
ax = fig.add_subplot(gs[1,0])
ax.plot(x,y2, '-b')
# using gridspec (or subplot2grid), existing axes
# object is apparently deleted
ax = fig.add_subplot(gs[0,0])
ax.plot(x,y2, '-r')

plt.show()

Answer 1

这实际上是一个微妙的错误，其中add_subplot如何确定一个轴是否存在。归结为这个事实：

In [220]: gs[0, 0] == gs[0, 0]
Out[220]: False

这是因为gridspec.__getitem__每次调用它都会返回一个新对象，SubplotSpec不会重载__eq__所以python检查'这是内存中的同一个对象'轴。

这就是错误，但我通过向__eq__添加SubplotSpec以及修补猴子matplotlib.gridspec.SubplotSpec来修复它的天真尝试不起作用（我没有时间）找出原因），但如果你添加

def __eq__(self, other):
    return all((self._gridspec == other._gridspec,
                self.num1 == other.num1,
                self.num2 == other.num2))

到class SubplotSpec(object): ~L380中的matplotlib/gridspec.py并从源代码重新安装按预期工作。

PR to fix this似乎打破了各种其他事情。

如何在Matplotlib中使用subplot2grid / gridspec检索现有的子图轴？

1 个答案: