用pandas和matplotlib写代码。csv的50个标题归为4类,按照标题末尾分别为A,B,C,D这4种来归类。 然后分别做这4类的折线图
时间: 2024-03-02 21:50:04 浏览: 67
Matplotlib入门笔记(一)——折线图的绘制
好的,下面是使用 pandas 和 matplotlib 来实现的代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取 CSV 文件,假设标题列为 'title',数值列为 'value'
df = pd.read_csv('data.csv')
# 根据标题末尾分类
df['category'] = df['title'].str[-1]
# 分类数据
categories = df.groupby('category')['value'].apply(list)
# 绘制折线图
fig, axs = plt.subplots(1, 4, figsize=(15, 5))
for i, cat in enumerate(categories.index):
axs[i].plot(categories[cat])
axs[i].set_title(cat)
plt.show()
```
首先使用 pandas 的 read_csv() 函数读取 CSV 文件,然后使用 str[-1] 来获取标题的末尾字符,从而进行分类。接着使用 groupby() 函数对分类后的数据进行分组,并将每个分类的数值转换为列表。最后,使用 Matplotlib 绘制四个折线图,每个图的标题为 A、B、C、D 中的一个。
阅读全文