python - 如何在文本文件中搜索字符串？

Question

我想检查一个字符串是否在一个文本文件中。如果是，则执行 X。如果不是，则执行 Y。但是，由于某种原因，此代码始终返回 True。任何人都可以看出什么问题吗？

def check():
    datafile = file('example.txt')
    found = False
    for line in datafile:
        if blabla in line:
            found = True
            break

check()
if True:
    print "true"
else:
    print "false"

Answer 1

你总是得到 True 的原因已经给出，所以我再提供一个建议:

如果你的文件不是太大，你可以把它读成一个字符串，然后直接使用它(比每行读取和检查行更容易而且通常更快):

with open('example.txt') as f:
    if 'blabla' in f.read():
        print("true")

另一个技巧:您可以通过使用 mmap.mmap() 来缓解可能的内存问题。创建一个使用底层文件的“类似字符串”的对象(而不是读取内存中的整个文件):

import mmap

with open('example.txt') as f:
    s = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)
    if s.find('blabla') != -1:
        print('true')

注意:在 python 3 中，mmap 的行为类似于 bytearray 对象而不是字符串，因此您使用 find() 查找的子序列必须是 bytes 对象而不是字符串，例如。 s.find(b'blabla'):

#!/usr/bin/env python3
import mmap

with open('example.txt', 'rb', 0) as file, \
     mmap.mmap(file.fileno(), 0, access=mmap.ACCESS_READ) as s:
    if s.find(b'blabla') != -1:
        print('true')

您还可以在 mmap 上使用正则表达式，例如，不区分大小写的搜索:if re.search(br'(?i)blabla', s):