我正在阅读一个文本文件,我想绘制这些列(y和z),但是如果x列的值是'out',则6点和4点的颜色应该是红色。按照这种逻辑,图中有6个点,但是4个点应该是蓝色,其他两个颜色应该是红色。有两个图,每个pragh有3点。我怎样才能做到这一点?
#-------input.dat---------
# x y z
# col 1 col 2 col 3
# res 5 5
# out 6 4
# res 7 3
import matplotlib.pyplot as plt
import numpy as np
data[] =[]
with open('input.dat', skip_header=2,'r') as f:
lines = f.readlines()
for line in lines[2:]:
data.append(list(line.split()))
y = (data[:, 2])
z = (data[:, 3])
------------------------
plt.plot(float(line[1]),b+)
for line in enumerate(data)
if line[0] == 'OUT'
plt.plot(float(line[1]),r+)
-----------------------
plt.subplot(211)
plt.plot(y, 'b+', label=" Y figure ")
plt.subplot(212)
plt.plot(z, 'b+', label=" Z figure ")
plt.show()