在 Python 中的 csv 文件中搜索单词/短语

Question

我有一个 csv 格式的推文数据库，看起来像这样 - screen of csv database - 我需要使用 Python 代码对这个文件执行以下任务：

在推文中搜索某些词/短语（推文的文本在 C 列中），如果推文有我要查找的这个词/短语，我需要将这条推文的整行写入一个新的.csv 文件

并且（如果可能）从旧的 csv 文件中删除这条推文，或者在没有它的情况下创建一条新推文。

我希望我说清楚了。

Answer 1

您可以使用 csv 库来读取文件并一次搜索每一行。

import csv

with open('out.csv', 'w', newline='') as output_file:
    writer = csv.writer(output_file, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL)

    with open('test.csv') as csv_file:
        csv_read = csv.reader(csv_file, delimiter=',')
        for row in csv_read:
            if "a" in row[2]:
                writer.writerow(row)

这里是 python 文档的链接：https://docs.python.org/3/library/csv.html。希望这会有所帮助。

编辑：如果您想要多个搜索词，请在列表推导中使用 any()。

import csv

with open('out.csv', 'w', newline='') as output_file:
    writer = csv.writer(output_file, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL)

    with open('input.csv') as csv_file:
        csv_read = csv.reader(csv_file, delimiter=',')
        for row in csv_read:
            search_terms = ["term1", "term2"]

            if any([term in row[2] for term in search_terms]):
                writer.writerow(row)