使用 Python 删除 CSV 文件中的行

Question

I want to delete a specific line that matches the argument and remove the line from the CSV file. I can't seems to find a solution.

The sample CSV data are as follows:

Apple, 1.0, 1.0
Banana, 2.1, 2.1
Carrot, 3.2, 3.2

Let's say I only want to remove the banana row, is is possible?

def delete_place(name):
    file = open('data/db.csv', 'r')
    reader = csv.reader(file)

    for row in reader:
        if name != row[0]:
            return 'Place not found.', 400
        else:
            # todo: delete line in csv
            return 'Place with name {} deleted.'.format(name), 200

Answer 1

You can read csv file into pandas dataframe and then apply selection/deletion operation. Later dump the modified dataframe into a csv file

import pandas as pd
df = pd.read_csv('data/db.csv')
df = df[df['1']!='Banana'] # I am assuming the csv file had no header, hence by default it used 1 as the header of first column
df.to_csv('data/db_modified.csv', header=False, index=False)