python - 如何删除颜色条

Question

我想从轴上删除颜色条，以便轴返回到其默认位置。

为了清楚起见，请看一下这段代码(或者更好地运行它):

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable

# Init figure
figure, axes = plt.subplots()
axes.set_aspect(1)

# Create some random stuff
image = axes.imshow(np.random.random((10, 10)))
plt.pause(2)

# Add a color bar to the axes
cax = make_axes_locatable(axes).append_axes("right", size="5%", pad=0.05)
colorBar = figure.colorbar(image, cax=cax)
colorBar.ax.tick_params(length=0, labelright=False)
plt.pause(2)

# Remove the color bar
colorBar.remove()

plt.show()

如您所见，会发生以下情况:在那一刻，我将颜色条添加到轴上，轴本身正在稍微改变其位置。它向左移动以为颜色条创造一些空间。在更复杂的情况下，例如在 Gridspec 中，它也可能会稍微改变其大小。

我想要得到的:在那一刻，我再次删除颜色条，我想让轴回到原来的位置(及其原始大小)。

当我删除颜色条时，这不会自动发生。轴仍位于左侧位置，而不是回到中心。 我如何实现这一目标？

我需要类似 make_axes_locateable 的倒数.

Answer 1

make_axes_locatable 将轴定位器设置为新定位器。要反转此步骤，您需要跟踪旧定位器 (original_loc = ax.get_axes_locator()) 并将其重置到轴 (ax.set_axes_locator(original_loc))删除颜色条后。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable

# Init figure
fig, ax = plt.subplots()
ax.set_aspect(1)

# Create some random stuff
image = ax.imshow(np.random.random((10, 10)))
plt.pause(2)

# Add a color bar to the axes
original_loc = ax.get_axes_locator()
cax = make_axes_locatable(ax).append_axes("right", size="5%", pad=0.05)
colorBar = fig.colorbar(image, cax=cax)
colorBar.ax.tick_params(length=0, labelright=False)
plt.pause(2)

# Remove the color bar
colorBar.remove()
ax.set_axes_locator(original_loc)
plt.pause(0.0001)

plt.show()