如何在python中更改多边形的特定边的颜色？

import matplotlib.pyplot as plt
import matplotlib.patches as patches

# 创建一个六边形对象
polygon = patches.Polygon([[0, 0], [1, 0], [1, 1], [0.5, 1.5], [0, 1], [0, 0]], closed=True)

# 创建绘图对象和子图
fig, ax = plt.subplots()

# 将多边形添加到子图中
ax.add_patch(polygon)

# 更改特定边的颜色（例如第三条边）
edge_index = 2
polygon.set_edgecolor('red')
polygon.set_path_effects([plt.patheffects.withStroke(linewidth=5, foreground='red')])

# 设置坐标轴范围
ax.set_xlim([-0.5, 1.5])
ax.set_ylim([-0.5, 1.5])

# 显示图形
plt.show()