извлекает содержимое файла csv в виде списков

Question

Каков наилучший и самый элегантный способ получить содержимое файла csv в виде списков? Обратите внимание, что требуемые столбцы, такие как Страна, Область и Население, должны быть идентифицированы автоматически, как в приведенной ниже программе.

with open ('data.csv','r') as infile: 
    contents = infile.read().split(',') 

with open ('data.csv','r') as infile: 
    countries = [item.split(',')[contents.index('Country')] for item in infile] 

with open ('data.csv','r') as infile: 
    areas = [item.split(',')[contents.index('Area')] for item in infile] 

with open ('data.csv','r') as infile: 
    populations = [item.split(',')[contents.index('Population')] for item in infile] 

print (countries) 
print (areas) 
print (populations) 

Выше программы работает, но ищет довольно короткую и читаемую программу.

Update: Чтобы предотвратить ошибку, когда некоторые колонки, такие как «страна» не существует в каком-то файле, программа может быть улучшена:

countries = [item.split(',')[contents.index('Country')] for item in infile if 'Country' in contents] 

Как это можно сделать с помощью модуля Csv ??

Answer 1

Используйте csv module для CSV файлов:

import csv 

countries = [] 
areas = [] 
populations = [] 

with open ('data.csv', newline='') as infile: 
    reader = csv.DictReader(infile) 
    for row in reader: 
     countries.append(row['Country']) 
     areas.append(row['Area']) 
     populations.append(row['Population']) 

где DictReader() класс автоматически использует первую строку заголовков в качестве ключей для остальной части файла.

Или используйте:

import csv 

with open ('data.csv', newline='') as infile: 
    reader = csv.reader(infile) 
    next(reader, None) # skip the header first 
    countries, areas, populations = zip(*reader) 

, если все, что вам есть 3 колонки, страна, область и население, в этом порядке.