python批量读取csv并写入,Python读取CSV文件，并写入其他跳过列

I have CSV input file with 18 columns

I need to create new CSV file with all columns from input except column 4 and 5

My function now looks like

def modify_csv_report(input_csv, output_csv):

begin = 0

end = 3

with open(input_csv, "r") as file_in:

with open(output_csv, "w") as file_out:

writer = csv.writer(file_out)

for row in csv.reader(file_in):

writer.writerow(row[begin:end])

return output_csv

So it reads and writes columns number 0 - 3, but i don't know how skip column 4,5 and continue from there

解决方案

This is what you want:

import csv

def remove_csv_columns(input_csv, output_csv, exclude_column_indices):

with open(input_csv) as file_in, open(output_csv, 'w') as file_out:

reader = csv.reader(file_in)

writer = csv.writer(file_out)

writer.writerows(

[col for idx, col in enumerate(row)

if idx not in exclude_column_indices]

for row in reader)

remove_csv_columns('in.csv', 'out.csv', (3, 4))