从文件中查找包含另一个文件内容的行Python

Question

所以..这就是我想要做的..对于数据文件中的每一行，请检查其他文件是否包含此字符串。

我尝试了其他帖子中的一些内容，但都不是一件好事。

下面的代码说，即使它们存在于文件中的某个位置，它也找不到任何要查找的字符串。

def search():
    file1= open('/home/example/file1.txt', 'r')
    datafile= open('/home/user/datafile.txt', 'r')

    for line in datafile:
        if line in file1:
            print '%s found' % line
        else:
            print '%s not found' % line

search()

Answer 1

假设第一个文件的内容不是很大，您可以将整个文件读取为字符串，然后使用字符串包含进行检查：

def search():
    file1_content = open('/home/example/file1.txt').read()
    datafile = open('/home/user/datafile.txt')

    for line in datafile:
        if line in file1_content:
            print '%s found' % line
        else:
            print '%s not found' % line

请注意， open的默认模式是'r' ，因此，如果您以文本模式阅读，则实际上不需要传递该参数。

Answer 2

您可以将文件读入一set ，然后检查是否包含在第二个文件中。 set的通常在检查包含列表时更快。

def search():
    file1 = set(open('/home/example/file1.txt'))
    datafile= open('/home/user/datafile.txt', 'r')

    for line in datafile:
        if line in file1:
            print '%s found' % line
        else:
            print '%s not found' % line

您还可以使用设置操作来提取例如不在第一个文件中的所有行：

set(open('/home/user/datafile.txt', 'r')) - set(open('/home/example/file1.txt'))