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

Question

I have a database of tweets in csv format which looks like this - screen of csv database - and I need to perform the following task with this file using Python code:

Search for certain words/phrases in tweets (text of a tweet is in the column C) and if the tweet has this word/phrase I'm looking for, I need to write the whole row with this tweet to a new csv file

and (if possible) to delete this tweet from the old csv file or create a new one without it.

I hope I made it clear.

Answer 1

You can use the csv library to read the file and do your search on each row one at a time.

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)

Here is a link to the python docs: https://docs.python.org/3/library/csv.html . Hope this helps.

EDIT: If you want more than one search term, use any() on a list comprehension.

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)