Python和CSV从用户选择的特定行读取数据

Question

我正在做一个读取CSV文件的python脚本。这是代码:

with open("DATABASE.csv", "r", newline="") as a:
csv_reader = csv.DictReader(a)
rows = list(csv_reader)
print(rows[2]["NAME"] rows[2]["SURNAME"])

[2]表示读取文件的第二行。

如何用用户选择的数字替换它?如果我将用户的选择保存到一个变量中，并将变量插入[ ]中，它会回复TypeError: list indices must be integers or slices, not str

,

还有一件小事:有没有办法删除CSV文件中的一行?

Answer 1

input()返回字符串，因此您必须将其强制转换为整数，使用int(input())更改输入代码。

如果您想删除csv文件中的一行，只需修改包含csv文件的列表，在您的情况下是rows变量，然后将其保存回csv文件。