Прочитайте конкретные столбцы из файла CSV с модулем CSV? ⇐ Python

[Anonymous]

Я пытаюсь проанализировать файл CSV и извлечь данные из только конкретных столбцов.

Код: Выделить всё

ID | Name | Address | City | State | Zip | Phone | OPEID | IPEDS |
10 | C... | 130 W.. | Mo.. | AL... | 3.. | 334.. | 01023 | 10063 |

Я пытаюсь забрать только конкретные столбцы, скажем, id , name , Zip и Phone .

код, на который я смотрел, привел меня к поверить, что я могу вызвать конкретный столбец, так что IE: Имя будет соответствует 2 создаст все элементы в столбце 2. Только это не так.import sys, argparse, csv
from settings import *

# command arguments
parser = argparse.ArgumentParser(description='csv to postgres',\
fromfile_prefix_chars="@" )
parser.add_argument('file', help='csv file to import', action='store')
args = parser.parse_args()
csv_file = args.file

# open csv file
with open(csv_file, 'rb') as csvfile:

# get number of columns
for line in csvfile.readlines():
array = line.split(',')
first_item = array[0]

num_columns = len(array)
csvfile.seek(0)

reader = csv.reader(csvfile, delimiter=' ')
included_cols = [1, 2, 6, 7]

for row in reader:
content = list(row for i in included_cols)
print content
< /code>

И я ожидаю, что это распечатает только конкретные столбцы, которые я хочу для каждой строки, за исключением этого, я получаю только последний столбец. < /p>

Подробнее здесь: https://stackoverflow.com/questions/165 ... csv-module