过滤文本文件中包含特定单词的行

Question

我想编写一个程序，过滤文本文件中包含单词“apple”的行，并将这些行写入新的文本文件中。

我所尝试的只是在新文本文件中写入“apple”一词，而我想要整行。

Answer 1

使用可以使用列表理解获取包含“apple”的所有行：

[ line for line in open('textfile') if 'apple' in line]

因此 - 同样在一个代码行中 - 您可以创建新的文本文件：

open('newfile','w').writelines([ line for line in open('textfile') if 'apple' in line])

并且 eyquem 是对的：将其保留为迭代器并写入肯定会更快

open('newfile','w').writelines(line for line in open('textfile') if 'apple' in line)

Answer 2

from itertools import ifilter

with open('source.txt','rb') as f, open('new.txt','wb') as g:
    g.writelines(ifilter(lambda line: 'apple' in line, f))

Answer 3

使用生成器，内存效率高且速度快，

def apple_finder(file):
    for line in file:
        if 'apple' in line:
             yield line


source = open('forest','rb')

apples = apple_finder(source)

我喜欢简单的解决方案，不会对阅读造成脑损伤:-)

Answer 4

对于Python3 - 这是有效且快速的示例

    text = b'line contains text'
    with open('input.txt', 'rb') as file_in:
        with open('output.txt', 'wb') as file_out:
            file_out.writelines(
                filter(lambda line: text in line, file_in)
            )

测试：

input.txt：

Test line contains text
Not line not contains this text

HEY
Another line contains text

output.txt：

Test line contains text
Another line contains text

有关代码的更多信息：

b'line contains text' - b< /code> 表示binary，我们对这种字符串进行操作，跳过一些编码等问题。
官方文档： https://docs.python.org /3/library/stdtypes.html?highlight=binary#bytes-objects

rb wb - 对 read 和 进行操作>使用二进制类似对象进行写入操作
官方文档： https://docs.python.org/3/library /io.html#binary-io

filter() - 采用表达式和可迭代对象。返回过滤后的对象。在我们的示例中，filter 获取所有行（可迭代对象），并为每行应用 lambda 通知 filter（如果给定） line 是否应该返回。

lambda - 包含两个元素参数：表达式。在我们的示例中，lambda 检查line 是否包含给定的文本。检查表达式后返回True或False。

使用 lambda 和 filter 的示例：https://blog.finxter.com/how-to-filter-in-python-using-lambda-functions/

Answer 5

if "apple" in line: 应该可以工作。