python提取csv文件几列并保存为csv

要提取csv文件的几列并保存为csv，可以使用以下代码：

import csv

# 要提取的列的索引
columns = [0, 3, 5]

# 读取原始csv文件
with open('原始文件.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)  # 获取表头
    
    # 获取要提取的列的表头
    header_new = [header[i] for i in columns]
    
    # 逐行读取数据并提取要保留的列
    rows_new = []
    for row in reader:
        row_new = [row[i] for i in columns]
        rows_new.append(row_new)
        
# 保存提取后的csv文件
with open('提取后的文件.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(header_new)
    writer.writerows(rows_new)

在这个代码中，首先定义了要提取的列的索引，然后使用csv模块读取原始csv文件，获取表头和数据。接着，根据要提取的列的索引，获取要保留的列的表头和数据，并保存到一个新的列表中。最后，使用csv模块将新的表头和数据写入到新的csv文件中。