Python читает конкретные столбцы CSV в список
У меня есть CSV-файл с 3 столбцами. Я хотел бы прочитать файл и поместить столбец № 2 в список.
В настоящее время он читает файл и помещает все три столбца в виде кортежей в список:
import csv
with open ('list.csv', 'rb') as f:
reader = csv.reader(f)
the_list = map(tuple, reader)
print the_list
output = [('1', 'bob', '23'), ('2', 'jane', '21')]
тогда как я хочу, чтобы выходной список был [('bob'), ('jane')], и я не уверен, как это сделать. Спасибо
2 ответа
Решение
Используйте понимание списка.
reader = csv.reader(f)
print [i[1] for i in reader]
import csv
with open('jjon.csv', 'rU') as infile:
reader = csv.reader(infile)
print [i[0] for i in reader]