从CSV动态删除列

Question

我想动态地从CSV中删除列，这是我迄今为止所做的。我不知道在哪里，从这里走，但：

# Remove column not needed. 
column_numbers_to_remove = 3,2, 
file = upload.filepath 
#I READ THE FILE 
file_read = csv.reader(file) 

REMOVE 3 and 2 column from the CSV 
UPDATE SAVE CSV 

Answer 1

使用enumerate拿到列索引，并创建一个新的行没有你不想要的列...如：

for row in file_read: 
    new_row = [col for idx, col in enumerate(row) if idx not in (3, 2)] 

然后使用csv.writer在某处写出您的行...

Answer 2

读取csv并在删除列后写入另一个文件。

import csv 
creader = csv.reader(open('csv.csv')) 
cwriter = csv.writer(open('csv2.csv', 'w')) 

for cline in creader: 
    new_line = [val for col, val in enumerate(cline) if col not in (2,3)] 
    cwriter.writerow(new_line)