python自制图例

可以使用matplotlib库中的plt.legend()函数来创建自定义图例。该函数可以自动创建带有标签属性的图表元素的图例。

以下是一个示例代码，演示如何使用plt.legend()函数创建自定义图例：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 1000)
fig, ax = plt.subplots()
ax.plot(x, np.sin(x), '-b', label='Sine')
ax.plot(x, np.cos(x), '--r', label='Cosine')
ax.axis('equal')

leg = ax.legend()

在这个例子中，我们首先创建了一个包含两条曲线的图表。然后，我们使用`label`参数为每条曲线指定一个标签。最后，我们调用`ax.legend()`函数来创建图例，并将其赋值给变量`leg`。图例将自动根据曲线的标签属性显示相应的图例项。