比循环整个文件更好的regex实现？

Question

我有这样的文件：

#     BJD     K2SC-Flux EAPFlux   Err  Flag Spline
2457217.463564 5848.004 5846.670 6.764 0 0.998291
2457217.483996 6195.018 6193.685 6.781 1 0.998291
2457217.504428 6396.612 6395.278 6.790 0 0.998292
2457217.524861 6220.890 6219.556 6.782 0 0.998292
2457217.545293 5891.856 5890.523 6.766 1 0.998292
2457217.565725 5581.000 5579.667 6.749 1 0.998292
2457217.586158 5230.566 5229.232 6.733 1 0.998292
2457217.606590 4901.128 4899.795 6.718 0 0.998293
2457217.627023 4604.127 4602.793 6.700 0 0.998293

我需要找到并计数标志为1的行。（第5栏）我是这样做的：

foundlines=[]
c=0
import re
with open('examplefile') as f:
    for index, line in enumerate(f):
        try:
            found = re.findall(r' 1 ', line)[0]
            foundlines.append(index)
            print(line)
            c+=1
        except:
            pass
print(c)

在Shell中，我只需执行grep " 1 " examplefile | wc -l，这比上面的Python脚本要短得多。python脚本可以工作，但我感兴趣的是，是否有比上面的脚本更短、更紧凑的方法来完成任务？我更喜欢Shell的简短，所以我希望在Python中有类似的东西。你知道吗

Answer 1

最短代码

在某些特定的前提下，这是一个非常简短的版本：

• 你只需要计算像grep调用这样的事件
• 保证每行只有一个" 1 "
• " 1 "只能出现在所需列中
• 你的文件很容易放进内存

请注意，如果不满足这些前提条件，这可能会导致内存问题或返回误报。你知道吗

print(open("examplefile").read().count(" 1 "))

简单多用，略长

当然，如果你以后真的对这些台词感兴趣，我建议你：

df = pandas.read_table('test.txt', delimiter=" ",
                       comment="#",
                       names=['BJD', 'K2SC-Flux', 'EAPFlux', 'Err', 'Flag', 'Spline'])

要获取标志为1的所有行：

flaggedrows = df[df.Flag == 1]

退货：

            BJD  K2SC-Flux   EAPFlux    Err  Flag    Spline
1  2.457217e+06   6195.018  6193.685  6.781     1  0.998291
4  2.457218e+06   5891.856  5890.523  6.766     1  0.998292
5  2.457218e+06   5581.000  5579.667  6.749     1  0.998292
6  2.457218e+06   5230.566  5229.232  6.733     1  0.998292

数一数：

print(len(flaggedrows))

返回4