我正在测试一种算法,我想使用 matplotlib 生成一系列显示中间结果的图形。
我不需要动画,也不需要屏幕上的多个数字,也不需要子图。
我只想生成一系列图形(可能使用 pyplot),完成后会显示一个窗口。然后我想使用箭头在数字序列中导航。
我怎么能做这样的事情?
我试图搜索,但我只能在屏幕上找到子图或多个数字。
谢谢
我正在测试一种算法,我想使用 matplotlib 生成一系列显示中间结果的图形。
我不需要动画,也不需要屏幕上的多个数字,也不需要子图。
我只想生成一系列图形(可能使用 pyplot),完成后会显示一个窗口。然后我想使用箭头在数字序列中导航。
我怎么能做这样的事情?
我试图搜索,但我只能在屏幕上找到子图或多个数字。
谢谢
最通用的方法是在同一个图中创建一系列轴,并且一次只显示一个。
这是一个示例(左右箭头键控制显示哪个图):
import matplotlib.pyplot as plt
import numpy as np
def main():
x = np.linspace(0, 10, 100)
axes = AxesSequence()
for i, ax in zip(range(3), axes):
ax.plot(x, np.sin(i * x))
ax.set_title('Line {}'.format(i))
for i, ax in zip(range(5), axes):
ax.imshow(np.random.random((10,10)))
ax.set_title('Image {}'.format(i))
axes.show()
class AxesSequence(object):
"""Creates a series of axes in a figure where only one is displayed at any
given time. Which plot is displayed is controlled by the arrow keys."""
def __init__(self):
self.fig = plt.figure()
self.axes = []
self._i = 0 # Currently displayed axes index
self._n = 0 # Last created axes index
self.fig.canvas.mpl_connect('key_press_event', self.on_keypress)
def __iter__(self):
while True:
yield self.new()
def new(self):
# The label needs to be specified so that a new axes will be created
# instead of "add_axes" just returning the original one.
ax = self.fig.add_axes([0.15, 0.1, 0.8, 0.8],
visible=False, label=self._n)
self._n += 1
self.axes.append(ax)
return ax
def on_keypress(self, event):
if event.key == 'right':
self.next_plot()
elif event.key == 'left':
self.prev_plot()
else:
return
self.fig.canvas.draw()
def next_plot(self):
if self._i < len(self.axes):
self.axes[self._i].set_visible(False)
self.axes[self._i+1].set_visible(True)
self._i += 1
def prev_plot(self):
if self._i > 0:
self.axes[self._i].set_visible(False)
self.axes[self._i-1].set_visible(True)
self._i -= 1
def show(self):
self.axes[0].set_visible(True)
plt.show()
if __name__ == '__main__':
main()
如果它们都是相同类型的情节,您可以更新相关艺术家的数据。如果每个图中的项目数量相同,这尤其容易。我暂时不举一个例子,但如果上面的例子太占用内存,那么仅仅更新艺术家的数据会轻得多。