用pandas和matplotlib写代码。csv有50列,先把50个标题归为4类,按照标题末尾分别为A,B,C,D这4种来归类。 然后分别做这4类的折线图
时间: 2024-03-02 09:50:07 浏览: 66
使用pandas模块读取csv文件和excel表格,并用matplotlib画图的方法
好的,下面是使用 pandas 和 matplotlib 来实现的代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取 CSV 文件
df = pd.read_csv('data.csv')
# 根据标题末尾分类
categories = {}
for col in df.columns:
cat = col[-1]
if cat not in categories:
categories[cat] = []
categories[cat].append(col)
# 分类数据
data = {}
for cat in categories:
data[cat] = df[categories[cat]]
# 绘制折线图
fig, axs = plt.subplots(1, 4, figsize=(15, 5))
for i, cat in enumerate(data):
axs[i].plot(data[cat])
axs[i].set_title(cat)
plt.show()
```
首先使用 pandas 的 read_csv() 函数读取 CSV 文件,然后遍历每个标题,根据末尾字符将其分类,并将分类后的标题存储在 categories 字典中。接着使用字典推导式和 df[] 操作符来获取每个分类的数据,并将其存储在 data 字典中。最后,使用 Matplotlib 绘制四个折线图,每个图的标题为 A、B、C、D 中的一个。
阅读全文