读取一个输入文件并绘制多个

Question

我正在尝试读取以下格式的输入文件。其中Col[1] 是 x 轴，Col[2] 是 y 轴，col[3] 是某个名称。我需要为col[3] 的单独名称绘制多个折线图。例如：名称sd 和x,y 值将有一个折线图，而名称gg 和相对x,y 值将有另一个折线图。所有在一个输出图像，但单独的图表。 Python Matplotlib 可以吗？请务必将我重定向到任何示例。我已经检查了找不到任何东西。

akdj 12:00 34515 sd
sgqv 13:00 34626 sd
dfbb 13:00 14215 gg
ajdf 13:30 14224 gg
dsfb 13:45 25672 FW 
sfhh 14:00 85597 ad

感谢您的宝贵建议

Answer 1

您可以使用条件z=='some tag' 来索引x 和y 数组

这是一个应该这样做的示例（基于your previous question 中的代码）。使用集合自动创建标签：

import csv
import datetime as dt
import numpy as np
import matplotlib.pyplot as plt

threshold = 30000
x,y,z = [],[],[]
csv_reader = csv.reader(open('input.csv'))

for line in csv_reader:
    y.append(int(line[2]))
    x.append(dt.datetime.strptime(line[1],'%H:%M'))
    z.append(line[3])

x=np.array(x)
y=np.array(y)

tags = list(set(z))
z=np.array(z)

fig=plt.figure()

for tag in tags:
    plt.plot(x[z==tag],y[z==tag],'o-',label=tag)

fig.autofmt_xdate()

plt.legend(loc=2)

plt.savefig('test.png')