如何使用python在CSV中僅刪除一行

Question

我正在嘗試制作一個程序來在CSV文件中存儲名稱列表，並且試圖添加一個刪除行的功能，該功能無法正常工作，因為它刪除了CSV文件中的所有內容。

我試過使用writer.writerow（row），但沒有用。

        memberName = input("Please enter a member's name to be deleted.")
        imp = open('mycsv.csv' , 'rb')
        out = open('mycsv.csv' , 'wb')
        writer = csv.writer(out)
        for row in csv.reader(imp):
            if row == memberName:
                writer.writerow(row)
        imp.close()
        out.close()

我希望程序僅刪除包含memberName的行，但是它將刪除CSV文件中的每一行。 如何將其更改為僅刪除一行？

Answer 1

您不能在讀取文件時寫入同一文件。 而是使用另一個文件進行輸出，例如：

import csv

member_name = input("Please enter a member's name to be deleted: ")

with open('in_file.csv') as in_file, open('out_file.csv', 'w') as out_file:
    reader = csv.reader(in_file)
    writer = csv.writer(out_file)
    for row in reader:
        if member_name not in row:  # exclude a specific row
            writer.writerow(row)

或者，您可以將所需的行存儲在內存中，並在重置文件指針之后將它們寫回到輸入文件中：

import csv

member_name = input("Please enter a member's name to be deleted: ")

with open('in_file.csv', 'r+') as in_file:
    reader = csv.reader(in_file)
    rows = [row for row in csv.reader(in_file) if member_name not in row]
    in_file.seek(0)
    in_file.truncate()
    writer = csv.writer(in_file)
    writer.writerows(rows)

Answer 2

這對我有用：您可以將csv文件的內容寫到列表中，然后在python中編輯列表，然后將列表寫回到csv文件中。

lines = list()
memberName = input("Please enter a member's name to be deleted.")
with open('mycsv.csv', 'r') as readFile:
    reader = csv.reader(readFile)
    for row in reader:
        lines.append(row)
        for field in row:
            if field == memberName:
                lines.remove(row)

with open('mycsv.csv', 'w') as writeFile:
    writer = csv.writer(writeFile)
    writer.writerows(lines)