python画图,根据标题名称以A,B,C,D结尾的归为一类,归类后按照4个类别画4个折线图
时间: 2024-03-02 19:49:58 浏览: 64
可以通过字符串的 endswith() 方法来判断标题名称是否以 A,B,C,D 结尾,然后将其归为相应的类别。
下面是一个简单的示例代码,假设有一个数据集 data,其中包含标题和对应的数值:
```python
import matplotlib.pyplot as plt
# 分类数据
categories = {'A': [], 'B': [], 'C': [], 'D': []}
# 读取数据
with open('data.txt', 'r') as f:
for line in f.readlines():
title, value = line.strip().split('\t')
if title.endswith('A'):
categories['A'].append(float(value))
elif title.endswith('B'):
categories['B'].append(float(value))
elif title.endswith('C'):
categories['C'].append(float(value))
elif title.endswith('D'):
categories['D'].append(float(value))
# 画图
fig, axs = plt.subplots(1, 4, figsize=(15, 5))
for i, cat in enumerate(categories):
axs[i].plot(categories[cat])
axs[i].set_title(cat)
plt.show()
```
在上面的代码中,我们首先定义了一个字典 categories,用于存储分类数据。然后读取数据集,对每个标题进行分类,并将对应的数值添加到相应的列表中。最后,使用 Matplotlib 画出四个折线图,每个图的标题为 A、B、C、D 中的一个。如果数据集较大,可以考虑使用 Pandas 库来读取和处理数据。
阅读全文