python - 基于词表过滤行

Question

你好我有这段代码可以过滤一个特定单词的所有行('test'，我想知道是否有人可以通过解释如何过滤不止一个单词的行来提供帮助，所以如果我有一个文件列出所有过滤词和源文件我将能够显示其中包含任何过滤词的所有源代码行。谢谢!

def cat(openfile):
  with open(openfile) as file:
    return file.read()

def getlinewith(filecontents, containing):
  for item in filecontents.split('\n'):
    if containing in item:
      yield item.strip()

matchedlines = []

for line in getlinewith(cat('C\\testdata_all.txt'), 'test'):
   print(line)
   matchedlines.append(line)

print(matchedlines)

Answer 1

使用 any :

def getlinewith(filecontents, containings):
  for item in filecontents.split('\n'):
    if any(containing in item for containing in containings):
      # `any` will return `True` as soon as it find a match
      yield item.strip()

matchedlines = []

for line in getlinewith(cat(r'C:\testdata_all.txt'), ['test', 'other_word']):
    ...