用pandas和matplotlib写代码。csv有50个标题,把50个标题分别按照标题末尾分别为(A),(B),(C),(D)的文本来归类。 然后分别做这4类的折线图
时间: 2024-03-02 09:51:18 浏览: 84
好的,我可以帮你写这段代码。首先你需要安装 pandas 和 matplotlib 这两个库,可以使用以下命令进行安装:
```
pip install pandas matplotlib
```
然后你需要将数据存储在一个 csv 文件中,并且每个数据都有一个标题。假设你的 csv 文件名为 data.csv,以下是代码实现:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取 csv 文件
data = pd.read_csv('data.csv')
# 根据标题末尾分类
grouped = {}
for col in data.columns:
key = col[-2]
if key not in grouped:
grouped[key] = pd.DataFrame()
grouped[key][col] = data[col]
# 分别绘制折线图
for key in grouped:
group = grouped[key]
group.plot(x='日期', y=group.columns, label=key)
plt.legend()
plt.show()
```
在这个例子中,我们使用了字典来存储分类后的数据,对于每一个标题,我们使用其末尾的字符作为 key,并将该列数据加入对应的 DataFrame 中。最后我们使用 plot 函数和循环来分别绘制每个分类的折线图。
需要注意的是,在绘制每个分类的折线图时,我们使用了 group.columns 来获取该分类下的所有列,并将其作为 y 轴的数据。如果你想要将所有的折线图放在同一张图中,只需要将 plt.show() 函数移动到循环外面即可。
希望这段代码可以帮到你,如果你还有其他问题,可以随时问我。
阅读全文