通过循环迭代更新条形图和绘制子图

时间:2016-03-07 12:33:39

标签: python loops matplotlib subplot

我写了以下代码片段,我正在尝试更新这些情节。 我得到的是新旧图块的重叠。 我研究了一下,发现我在当前轴上需要relim()autoscale_view(True,True,True)。 我仍然无法得到理想的行为。 有没有办法在调用plt.draw()之前强制pyplot删除/删除旧图形?

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

plt.ion()
a = np.arange(10)

fig,ax = plt.subplots(2,1)
plt.show()

for i in range(100):
    b = np.arange(10) * np.random.randint(10)
    ax[0].bar(a,b,align='center')
    ax[0].relim()
    ax[0].autoscale_view(True,True,True)
    ax[1].plot(a,b,'r-')
    ax[1].relim()
    ax[1].autoscale_view(True,True,True)
    plt.draw()
    time.sleep(0.01)
    plt.pause(0.001)

output image

2 个答案:

答案 0 :(得分:1)

Axes有一个方法clear()来实现这一目标。

for i in range(100):
    b = np.arange(10) * np.random.randint(10)

    ax[0].clear()
    ax[1].clear()

    ax[0].bar(a,b,align='center')
    # ...

Matplotlib Axes Documentation

relim()会始终根据新数据调整尺寸,以便获得静态图像。相反,我会使用set_ylim([min, max])来设置值的修复区域。

答案 1 :(得分:1)

无需重置轴限制或使用relim,您可能只想更新条形高度。

import numpy as np
import matplotlib.pyplot as plt

plt.ion()
a = np.arange(10)

fig,ax = plt.subplots(2,1)
plt.show()

b = 10 * np.random.randint(0,10,size=10)
rects = ax[0].bar(a,b, align='center')
line, = ax[1].plot(a,b,'r-')
ax[0].set_ylim(0,100)
ax[1].set_ylim(0,100)

for i in range(100):
    b = 10 * np.random.randint(0,10,size=10)
    for rect, h in zip(rects, b):
        rect.set_height(h)
    line.set_data(a,b)
    plt.draw()
    plt.pause(0.02)