在加载的csv的“n”位置添加一行

Question

我没能做到。但最让我困惑的是如何在csv的第n行中添加元素，例如，我想在csv的第2行中添加一行。你知道吗

mycsv.csv

name,last name
yeison, smith
lola, boa
elmo, spitia
anderson, exneider
juan, ortega

这是我的密码：

with open('mycsv.csv', 'w') as f:
 #I need add "barney, cubides" on position [2] of my csv
 f.write("barney, cubides") #not works properly..

你怎么能做到？你知道吗

Answer 1

你必须在读完文件后再写。因此，请阅读整个csv文件，并将每一行保存为一个列表，在需要的地方插入新行，然后重新编写整个文件。你知道吗

index_to_insert = 2
new_csv = []
new_line = "barney, cubides\n"
with open("mycsv.csv", "r") as f:
    new_csv = f.readlines()
new_csv.insert(index_to_insert, new_line)
with open("mycsv.csv", "w") as f:
    for line in new_csv:
        f.write(line)

另外，你可能想去掉csv文件中逗号前后的空白。你知道吗