Matplotlib -如何将标注添加到图例

import matplotlib.pyplot as plt

# 创建图表
fig, ax = plt.subplots()

# 绘制数据
x = [1, 2, 3, 4, 5]
y = [1, 4, 9, 16, 25]
ax.plot(x, y, label='y = x^2')

# 添加标注
handles, labels = ax.get_legend_handles_labels()
extra_handles = [plt.Line2D([0], [0], marker='o', color='red', markersize=5, linestyle=''),
                 plt.Line2D([0], [0], marker='s', color='blue', markersize=5, linestyle='')]
extra_labels = ['Point 1', 'Point 2']
handles.extend(extra_handles)
labels.extend(extra_labels)

# 创建图例
ax.legend(handles, labels)

# 显示图表
plt.show()