如何在 csv 文件中删除带有 Python 的单行

Question

我有一段代码试图获取用户的输入并删除 csv 文件中该特定行的行。我已经厌倦了使用下面的代码，但它也删除了行和 header。我只想删除行而不是 header，问题出在 remove_a_team 函数中。我怎样才能解决这个问题？谢谢。

import csv
import time

header_teams = ['TeamName', 'Member1', 'Member2', 'Member3', 'Member4', 'Member5']

#creating a header for the team's csv file.
with open('teams.csv', 'w', newline = '', encoding='UTF8') as f:
    writer = csv.writer(f)
    #writing the header for the csv file.
    writer.writerow(header_teams)
f.close()

def add_a_team():
    team_adder = []
    name = str(input("You are now entering the data for a team.\nEnter the team's name."))
    team_adder.append(name)
    time.sleep(0.3)
    name = str(input("Enter the first team member's name."))
    team_adder.append(name)
    time.sleep(0.3)
    name = str(input("Enter the second team member's name."))
    team_adder.append(name)
    time.sleep(0.3)
    name = str(input("Enter the third team member's name."))
    team_adder.append(name)
    time.sleep(0.3)
    name = str(input("Enter the fourth team member's name."))
    team_adder.append(name)
    time.sleep(0.3)
    name = str(input("Enter the fifth team member's name."))
    team_adder.append(name)
    time.sleep(0.3)
    print("Team successfully added.")
    print("\n")
    with open('teams.csv', 'a', newline = '', encoding='UTF8') as f:
        writer = csv.writer(f)
        #writing the data for each team.
        writer.writerow(team_adder)
    f.close()

def remove_a_team():
    content_team = input(str("What team would you like to remove?"))
    with open('teams.csv', 'r', newline = '', encoding='UTF8') as f:
        lines = f.readlines()
                         
    with open('teams.csv', 'w', newline = '', encoding='UTF8') as f:
        for line in lines:
            if line == content_team:
                writer.writerow(row)
            else:
                print("That team does not exist")
    f.close()

add_a_team()
remove_a_team()

Answer 1

猜测content_team = input(str("What team would you like to remove?"))应该给出你要删除的队名，那么在这块：

with open('teams.csv', 'w', newline = '', encoding='UTF8') as f:
        for line in lines:
            if line == content_team:
                writer.writerow(line)
            else:
                print("That team does not exist")

变量row不存在。我猜它应该是 line 而你打算写信给 f。你的逻辑倒置了。如果团队名称匹配，则您正在写该行，但希望相反。此外，您正在根据 content_team 检查整行，而不是仅检查第一个字段。整个函数应该是：

def remove_a_team():
    content_team = input(str("What team would you like to remove?"))
    with open('teams.csv', 'r', newline = '', encoding='UTF8') as f:
        lines = f.readlines()
                         
    found_team=False
    with open('teams2.csv', 'w', newline = '', encoding='UTF8') as f:
        for line in lines:
            if line.split(',')[0] == content_team:
                found_team=True
            else:
                f.write(line)
    if not found_team:                
        print("That team does not exist")
    f.close()

line.split(',')[0] 是否只给我们每行的第一个字段，我们将其与用户的输入进行比较

如何在 csv 文件中删除带有 Python 的单行

How can I delete a single line with Python in a csv file

python

csv