如何修改Matplotlib Collection的Segment？替代删除重建方案

问题背景 #

在使用Matplotlib的Collection组件时，需要修改Segment的取值，但不清楚如何直接操作现有Collection。当前做法是先移除旧Collection，再创建并添加新的Collection，代码如下：

# 当前使用方式
new_collections = {collection list}
for i in ax.collections: i.remove()
for i in new_collections: ax.add_collection(i)
ax.figure.canvas.draw()

希望通过直接修改现有Segment来提升效率，询问该操作是否可行。

可行方案 #

直接修改现有Collection的Segment完全可行，且效率远高于删除重建（尤其数据量大时）。不同类型的Collection有对应的属性可直接修改，以下是常见场景的操作方式：

• LineCollection：通过set_segments()方法更新线段数据，修改后调用局部更新方法即可，无需全量重绘画布：

# 假设已存在line_collection实例
line_collection.set_segments(new_segments)
# 高效触发画布更新，仅重绘变化区域
ax.figure.canvas.draw_idle()

• PolyCollection：修改多边形顶点可使用set_verts()方法，同样配合局部更新：

poly_collection.set_verts(new_vertices)
ax.figure.canvas.draw_idle()

• PathCollection：针对路径类集合，修改paths属性后触发局部更新：

path_collection.set_paths(new_paths)
ax.figure.canvas.draw_idle()

注意事项 #

• 局部更新优先用draw_idle()，它会在系统空闲时完成重绘，比全量的draw()更高效；
• 若用于动画场景，配合blit=True的动画模式能进一步优化性能。

内容的提问来源于stack exchange，提问作者white.seolpyo.com