DataFrame的列名怎么作为图例
时间: 2023-11-03 13:59:20 浏览: 56
Pandas DataFrame Notes
可以使用`matplotlib`或者`seaborn`库来绘制图表,并将DataFrame的列名作为图例。下面是一个使用`matplotlib`库来绘制折线图,并将DataFrame的列名作为图例的示例代码:
```python
import matplotlib.pyplot as plt
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({'x': [1, 2, 3, 4], 'y1': [10, 15, 13, 17], 'y2': [8, 12, 16, 11]})
# 设置x轴和y轴数据
x = df['x']
y1 = df['y1']
y2 = df['y2']
# 绘制折线图
plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
# 设置图例
plt.legend()
# 显示图表
plt.show()
```
在上面的代码中,我们使用`plt.plot`方法绘制折线图,并在每个`plot`方法中设置一个`label`参数,该参数的值为DataFrame中对应列的名称。然后,我们使用`plt.legend`方法将图例显示在图表中。
阅读全文