Вывод определенных столбцов из csv

#python #python_3x #csv

Есть сsv файл с именованными колонками вида:

id;name;address;zip
1;vasya;moscow;11111
2;oleg;sochi;22222


Как мне с помощью стандартного модуля csv вывести только столбцы к примеру name и
zip? В дальнейшем необходимо сформировать новый csv файл только с этими двумя колонками.
    

Ответы

Ответ 1

Решил, может кому пригодится:

import csv

with open('f_name', 'r') as f:
    incl_col = [1, 3]                          # индексы нужных столбцов
    new_csv = []                              # новый список для нового файла
    reader = csv.reader(f, delimiter=";")
    for row in reader:
        col = list(row[i] for i in incl_col)
        print(col)                            # вывод нужных столбцов
        new_csv.append(col)                   # заполняем новый список нужными столбцами

with open('new.csv', 'w') as f:               # создаем новый файл
    writer = csv.writer(f, delimiter=";")
    writer.writerows(new_csv)