python添加图例代码

在Python中，使用Matplotlib库可以很方便地添加图例。以下是一个示例代码，展示了如何在不同类型的图中添加图例：

import numpy as np
import matplotlib.pyplot as plt

# Example 1: Line plot
x = np.linspace(0, 2, 100)
plt.figure()
plt.plot(x, x+2, label='linear')
plt.plot(x, x**2, label='quadratic')
plt.plot(x, x**3, label='cubic')
plt.xlabel('x label')
plt.ylabel('y label')
plt.title("Legend Example 1")
plt.legend()

# Example 2: Subplot
x = np.linspace(0.0, 5.0)
y = x*x
plt.subplot(2, 1, 2)
plt.plot(x, y, 'g.-',label='quadratic')
plt.plot(x,x, 'r.-', label='linear')
plt.title('Legend Example 2')
plt.xlabel('numbers')
plt.ylabel('Square')
plt.legend()

# Example 3: Bar graph
plt.figure()
plt.barh(np.arange(26), np.random.randint(0,50,26), alpha = 0.5, color='r', label='Legend Bar Graph')
plt.title('Horizontal Bar Graph : Random')
plt.legend()

plt.show()

在这个示例代码中，我们使用了`plt.legend()`函数来添加图例。在每个示例中，我们都为不同的数据系列添加了标签，并使用`label`参数来指定它们。然后，我们使用`plt.legend()`函数来自动创建图例。在第一个示例中，我们没有指定位置参数，因此Matplotlib会自动选择最佳位置来放置图例。在第二个和第三个示例中，我们使用了默认的位置参数，将图例放置在右上角。