循环绘图（使用底图和 pyplot）....pyplot.clf() 的问题

Question

我正在为一个研究项目绘制一些天气数据。该图由 18 个时间步组成。我认为完成此任务的最佳方法是为每个时间步长创建一个新图，将其保存为文件，然后为下一个时间步长创建一个新图（使用 for 循环）。

例如：

map_init  #[Basemap Instance]
extra_shapes  #[Basemap.readshapefile object]

for i in range(timesteps):
    #plot the weather data for current timestep to current plot
    map_init.imshow(data[i])

    # extra_shapes are county boundaries.  Plot those as polygons
    pyplot.Polygon(map_init.extra_shapes[i])

    # Plot the state boundaries (in basemap)
    map_init.drawstates()

    # add a colorbar
    pyplot.colorbar()

    # Save the figure
    pyplot.savefig(filepath)

    #close figure and loop again (if necessary)
    pyplot.clf()

问题在于 pyplot.clf()

该代码除了一件事之外都可以工作。只有第一个情节按预期出现。随后的每个绘图都缺少 extra_shapes（即没有县边界）。我不明白 pyplot.clf() 的存在与 pyplot.Polygon() 的失败之间的关系？

如果删除，则会绘制 extra_shapes，但每个图都有多个颜色条（取决于 i 的值）。 pyplot.clf() 存在的唯一原因是避免最终绘图中出现 18 个颜色条。有没有一种方法可以强制每个图只有一个颜色条？

Answer 1

尝试创建一个新图形而不是使用 clf()。

例如

for i in range(timesteps):
    fig = pyplot.figure()
    ...
    fig.savefig(filepath)

，或者（并且更快）您可以只更新图像对象中的数据
（由 imshow() 返回）。

例如（完全未经测试）：

map_init  #[Basemap Instance]
extra_shapes  #[Basemap.readshapefile object]


#plot the weather data for current timestep to current plot
img = map_init.imshow(data[0])

# extra_shapes are county boundaries.  Plot those as polygons
plygn = pyplot.Polygon(map_init.extra_shapes[0])

# Plot the state boundaries (in basemap)
map_init.drawstates()

# add a colorbar
pyplot.colorbar()

for i in range(timestamps):
    img.set_data(data[i])
    plygn.set_xy(map_init.extra_shapes[i])
    pyplot.draw()
    pyplot.savefig(filepath)

但是，该方法可能无法很好地与底图配合使用。我也可能记错了重绘图形的正确方法，但我相当确定它只是 plt.draw()...

希望能有所帮助

编辑：刚刚注意到你正在循环内绘制多边形，还有。更新了第二个示例以正确反映这一点。