DataFrame的列名怎么作为图例
时间: 2023-11-03 17:07:10 浏览: 199
Python 借助DataFrame的pivot进行数据的可视化
你可以使用`matplotlib`或者`seaborn`库来绘制图表,并将DataFrame的列名作为图例。以下是一个使用`seaborn`库的例子:
```python
import seaborn as sns
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'x': [1, 2, 3, 4], 'y1': [10, 20, 30, 40], 'y2': [20, 30, 10, 40]})
# 使用seaborn绘制折线图
sns.lineplot(data=df.set_index('x'))
# 设置图例
import matplotlib.pyplot as plt
plt.legend(title='Legend', labels=df.columns[1:])
plt.show()
```
在这个例子中,我们首先创建了一个DataFrame,然后使用`set_index()`方法将`x`列设置为索引。之后,我们使用`lineplot()`函数绘制折线图,并使用`plt.legend()`方法将DataFrame的列名设置为图例。最后,我们使用`plt.show()`方法显示图表。
阅读全文