把CSV按类别分组,并画图。
时间: 2024-03-02 08:52:37 浏览: 96
好的,以下是示例代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取CSV文件
df = pd.read_csv('data.csv')
# 根据标题名字末尾的文本来归类
grouped = df.groupby(df.columns.str.split('(').str[1].str[:-1])
# 循环绘制折线图
for name, group in grouped:
group.plot(kind='line')
# 显示图例和标题
plt.legend()
plt.title('不同类别的折线图')
# 显示图形
plt.show()
```
其中,`data.csv`是你的CSV文件名字。这段代码会将标题末尾是`(apple)`、`(Pear)`、`(grape)`的标题分别归为一类,分别绘制3个折线图,并在图形上显示图例和标题。你可以根据实际情况修改代码,例如添加`x轴列名`和`y轴列名`等参数。
阅读全文