python取csv固定多列重写csv文件
时间: 2023-06-08 11:07:31 浏览: 85
python3,csv多列提取重新生成csv
好的,针对您的问题,我可以提供以下Python代码实现:
```
import csv
# 定义需要读取的列索引,这里选择1, 3和4列
columns = [1, 3, 4]
# 定义输入和输出文件路径
input_file = 'input.csv'
output_file = 'output.csv'
# 读取输入文件,并仅保留指定列
with open(input_file) as csv_file, open(output_file, 'w', newline='') as output:
reader = csv.reader(csv_file)
writer = csv.writer(output)
for row in reader:
output_row = [row[i] for i in columns]
writer.writerow(output_row)
```
在上面的代码中,我们首先定义要读取的列索引,然后指定输入和输出文件的路径。接下来,使用Python的csv模块读取输入文件,并仅保留指定列。最后,将结果写入输出文件中。此外,我们还指定了`newline`参数,这是为了避免在Windows系统上出现行结束符问题。
希望这可以回答您的问题。如果您需要进一步的帮助,请随时告诉我。
阅读全文