使用 Pyplot 绘制特定值的特定线条

Question

我正在尝试绘制两个此类数据文件：

name1.fits 0 0 2.40359218172
name2.fits 0 0 2.15961244263

第三列的值从 0 到 5。我想绘制第 2 列与第 4 列的对比图，但是，对于第 3 列中值小于 2（0 和 1）的行，我想将第 2 列移动-0.1，对于值大于 3（4 和 5）的行，我想将 col 2 移动 +0.1。

但是我的代码似乎将所有值都移动了 +0.1。这是我目前所拥有的：

import matplotlib.pyplot as plt
import numpy as np

with open('file1.txt') as data, open('file2.txt') as stds:

    lines1 = data.readlines()
    lines2 = stds.readlines()

    x1a = []
    x2a = []
    x1b = []
    x2b = []
    x1c = []
    x2c = []
    y1a = []
    y2a = []
    y1b = []
    y2b = []
    y1c = []
    y2c = []

    for line1 in lines1:
        p = line1.split()
        if p[2] < 2:        
            x1a.append(float(p[1]))
            y1a.append(float(p[3]))
        elif 1 < p[2] < 4:        
            x1b.append(float(p[1]))
            y1b.append(float(p[3]))
        elif p[2] > 3:        
            x1c.append(float(p[1]))
            y1c.append(float(p[3]))

    for line2 in lines2:
        q = line2.split()
        if q[2] < 2:        
            x2a.append(float(q[1]))
            y2a.append(float(q[3]))
        elif 1 < q[2] < 4:        
            x2b.append(float(q[1]))
            y2b.append(float(q[3]))
        elif q[2] > 3:        
            x2c.append(float(q[1]))
            y2c.append(float(q[3]))

    x1a = np.array(x1a)
    x2a = np.array(x2a)
    x1b = np.array(x1b)
    x2b = np.array(x2b)
    x1c = np.array(x1c)
    x2c = np.array(x2c)
    y1a = np.array(y1a)
    y2a = np.array(y2a)
    y1b = np.array(y1b)
    y2b = np.array(y2b)
    y1c = np.array(y1c)
    y2c = np.array(y2c)

minorLocator = AutoMinorLocator(5)

fig, ax = plt.subplots(figsize=(8, 8))
fig.subplots_adjust(left=0.11, right=0.95, top=0.94)

plt.plot(x1a-0.1,y1a,'b^',mec='blue',label=r'B0',ms=8)
plt.plot(x2a-0.1,y2a,'r^',mec='red',fillstyle='none',mew=0.8,ms=8)
plt.plot(x1b,y1b,'bo',mec='blue',label=r'B0',ms=8)
plt.plot(x2b,y2b,'ro',mec='red',fillstyle='none',mew=0.8,ms=8)
plt.plot(x1c+0.1,y1c,'bx',mec='blue',label=r'B0',ms=8)
plt.plot(x2c+0.1,y2c,'rx',mec='red',fillstyle='none',mew=0.8,ms=8)

plt.axis([-1.0, 3.0, 0., 4])
ax.xaxis.set_tick_params(labeltop='on')
ax.yaxis.set_minor_locator(minorLocator)

plt.show()

剧情如下：

plot

我很确定问题出在我的 "ifs" 上。我希望你能扫清道路 and/or 告诉我一个更好的选择。

Answer 1

当您进行查询 (if) 时，您必须确保转换发生在问题之前，因此：

   for line1 in lines1:
        p = line1.split()
        if p[2] < 2:        
            x1a.append(float(p[1]))
            y1a.append(float(p[3]))
        elif 1 < p[2] < 4:        
            x1b.append(float(p[1]))
            y1b.append(float(p[3]))
        elif p[2] > 3:        
            x1c.append(float(p[1]))
            y1c.append(float(p[3]))

，实际上应该是：

for line1 in lines1:
    p = line1.split()
    if float(p[2]) < 2:     # changed here   
        x1a.append(float(p[1]))
        y1a.append(float(p[3]))
    elif 1 < float(p[2]) < 4:  # There seems to be a problem with this if      
        x1b.append(float(p[1]))
        y1b.append(float(p[3]))
    elif float(p[2]) > 3:     # changed here    
        x1c.append(float(p[1]))
        y1c.append(float(p[3]))

您的 q 变量也是如此。另请注意，询问 1 < x < 4 将拦截 x > 3 和 x < 2。你也应该纠正这个。

使用 Pyplot 绘制特定值的特定线条

Plot specific lines for specific values with Pyplot

matplotlib

readlines

python-2.7