python - 提取特定的文本行？

Question

我有一个几十万行的大文本文件。我必须随机抽取文本文件中的 30,000 行特定行。这是我必须一次提取一行的程序:

big_file = open('C:\\gbigfile.txt', 'r')
small_file3 = open('C:\\small_file3.txt', 'w')
for line in big_file:
   if 'S0414' in line:
      small_file3.write(line)
gbigfile.close()
small_file3.close()

我怎样才能加快我需要查找的 30,000 行的速度>？

Answer 1

啊哈!因此，您真正的问题是如何在每行中测试多个条件，如果满足其中一个条件，则输出该行。我认为最简单的方法是使用正则表达式:

import re
keywords = ['S0414', 'GT213', 'AT3423', 'PR342'] # etc - you probably get those from some source
pattern = re.compile('|'.join(keywords))

for line in inf:
    if pattern.search(ln):
        outf.write(line)