Python中创建Polygon句柄数组的技术实现问题

报错原因分析 #

• plt.fill()返回的是包含单个Polygon对象的列表，而非单个数值。
• 你初始化的poly = np.zeros(n_poly)是float类型的numpy数组，无法存储列表或Python对象，因此抛出ValueError: setting an array element with a sequence。

正确的Polygon句柄存储方案 #

不用numpy数组，改用普通Python列表存储Polygon对象，同时推荐直接通过matplotlib.patches.Polygon创建对象（比plt.fill更易管理）：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Polygon

n_poly = 10
poly_list = []  # 用普通列表存储Polygon句柄

fig = plt.figure(frameon=False)
fig.set_size_inches(2.0, 2.0)
ax = plt.axes(xlim=(0, 10), ylim=(0, 10))

# 定义基础多边形顶点（正方形）
base_vertices = np.array([[0, 0], [1, 0], [1, 1], [0, 1]])

for i in range(n_poly):
    # 创建Polygon对象，初始位置平移i个单位
    poly = Polygon(base_vertices + [i, i], closed=True, facecolor='blue', alpha=0.5)
    ax.add_patch(poly)
    poly_list.append(poly)  # 存入列表，后续可通过poly_list[i]访问

plt.show()

高效动画实现（复用Polygon） #

利用FuncAnimation，每帧仅更新Polygon的顶点数据，避免重复创建和清绘，大幅降低计算量：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Polygon
from matplotlib.animation import FuncAnimation

n_poly = 10
poly_list = []
fig = plt.figure(frameon=False)
fig.set_size_inches(5.0, 5.0)
ax = plt.axes(xlim=(0, 15), ylim=(0, 15))

# 基础正方形顶点
base_vertices = np.array([[0,0], [1,0], [1,1], [0,1]])

# 初始化所有多边形
for i in range(n_poly):
    poly = Polygon(base_vertices + [i, i], closed=True, facecolor=f'C{i}', alpha=0.7)
    ax.add_patch(poly)
    poly_list.append(poly)

# 动画更新函数：仅修改顶点数据，不重建对象
def update(frame):
    for idx, poly in enumerate(poly_list):
        # 计算每帧的偏移量（示例：正弦曲线平移）
        offset = np.array([np.sin(frame/10 + idx*0.5), np.cos(frame/10 + idx*0.5)])
        # 更新多边形顶点
        poly.set_xy(base_vertices + [idx, idx] + offset)
    return poly_list

# 创建动画，blit=True只重绘变化区域，提升效率
ani = FuncAnimation(fig, update, frames=100, interval=50, blit=True)
plt.show()

关键优化点 #

• 对象复用：仅初始化一次Polygon，后续仅更新顶点数据，避免重复创建/销毁对象的开销。
• Blit优化：blit=True让Matplotlib只重绘变化的部分，而非整个画布，大幅提升动画帧率。
• 直接操作Polygon：用Polygon类直接创建对象，比plt.fill更易获取和管理句柄。

内容的提问来源于stack exchange，提问作者user20787420