如何在文本文件中搜索包含特定單詞的行，然后使用“找到”行創建一個新文件

Question

我需要在一些數據（文本文件）中搜索包含特定單詞的行，然后創建一個僅包含“找到”行的新文本文件。

例如，如果原始文本文件 (data.txt) 中的數據是：

Child 9  60lbs  Female Diabetes
Adult 25 180lbs Male   ADHD
Adult 46 200lbs Female epilepsy
Child 10 65lbs  Female ADHD

我要搜索關鍵字'Child'，新的文本文件（output.txt）將是：

Child 9  60lbs  Female Diabetes
Child 10 65lbs  Female ADHD  

到目前為止，這是我的代碼，我真的不知道如何將找到的行寫入新文件。

def main():
    
    Word1 = 'Child'
    
    #open an existing file. 
    OriginalData = open('recordData.txt', 'r')
    datalist = []

    for line in OriginalData.readlines():
            if Word1 in line: 
                #write the line to new file.
    
if __name__=="__main__":
    main()

Answer 1

search_word = 'your search word here'
with open('data.txt') as file:  # opens it, then closes it automatically at the end of the with block
    lines = [line for line in file.read().split('\n') if search_word in line]

with open('output.txt', 'w+') as output_file:  # 'w+' specifies it can be written to and created
    output_file.write('\n'.join(lines))

現在我們可以分解lines = [line for line in file.read().split('\n') if search_word in line]

file.read()返回整個文件的字符串

.split('\n')將字符串轉換為列表，在每個換行符處打破它（ '\n'是換行符）

if search_word in line所以它只添加帶有單詞的行

'\n'.join(lines)將選定的行重新組合在一起，然后使用write將其寫入文件