如何在 Python 绘图中添加图例

在 Python 中使用各种绘图库进行数据可视化时，添加图例是一个非常有用的操作，可以帮助读者更好地理解数据。下面介绍在 Python 绘图中添加图例的方法，主要包括 Matplotlib 和 Seaborn 两个库。

Matplotlib

Matplotlib 是 Python 中最常用的绘图库之一，支持多种类型的图表。添加图例的方法如下：

1. 在 plot 函数中加入 label 参数

import matplotlib.pyplot as plt

x = [1, 2, 3, 4]
y = [10, 8, 6, 2]

plt.plot(x, y, label='Line 1')

plt.legend()

plt.show()

2. 自定义图例

import matplotlib.pyplot as plt

x = [1, 2, 3, 4]
y1 = [10, 8, 6, 2]
y2 = [2, 6, 8, 10]

plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')

plt.legend(['First', 'Second'])

plt.show()

Seaborn

Seaborn 是一个基于 Matplotlib 的高级数据可视化库，它提供了一些便捷的函数和内置的主题，可用于制作更美观的图形。添加图例的方法如下：

1. 在函数中加入 label 参数

import seaborn as sns

tips = sns.load_dataset('tips')

sns.scatterplot(x='total_bill', y='tip', hue='smoker', data=tips, label='Smoker')

plt.legend()

plt.show()

2. 自定义图例

import seaborn as sns

tips = sns.load_dataset('tips')

sns.scatterplot(x='total_bill', y='tip', hue='smoker', data=tips)

plt.legend(title='Smoking?', loc='upper left', labels=['Yes', 'No'])

plt.show()

以上就是在 Python 绘图中添加图例的方法，可以根据不同的绘图需求灵活运用。