У меня есть файл csv
col1, col2, col3
1, 2, 3
4, 5, 6
Я хочу создать список словарей из этого csv.
вывод:
a= [{'col1':1, 'col2':2, 'col3':3}, {'col1':4, 'col2':5, 'col3':6}]
Как я могу это сделать?
У меня есть файл csv
col1, col2, col3
1, 2, 3
4, 5, 6
Я хочу создать список словарей из этого csv.
вывод:
a= [{'col1':1, 'col2':2, 'col3':3}, {'col1':4, 'col2':5, 'col3':6}]
Как я могу это сделать?
Используйте csv.DictReader
:
>>> import csv
>>>
>>> with open('test.csv') as f:
... a = [{k: int(v) for k, v in row.items()}
... for row in csv.DictReader(f, skipinitialspace=True)]
...
>>> a
[{'col2': 2, 'col3': 3, 'col1': 1}, {'col2': 5, 'col3': 6, 'col1': 4}]
Используя модуль csv
и понимание списка:
import csv
with open('foo.csv') as f:
reader = csv.reader(f, skipinitialspace=True)
header = next(reader)
a = [dict(zip(header, map(int, row))) for row in reader]
print a
Вывод:
[{'col3': 3, 'col2': 2, 'col1': 1}, {'col3': 6, 'col2': 5, 'col1': 4}]
Другой более простой ответ:
import csv
with open("configure_column_mapping_logic.csv", "r") as f:
reader = csv.DictReader(f)
a = list(reader)
print a
Ну, в то время как другие люди делали это разумно, я реализовал это наивно. Я полагаю, что мой подход имеет преимущество в том, что он не нуждается в каких-либо внешних модулях, хотя он, вероятно, потерпит неудачу с странными конфигурациями значений. Здесь он просто для справки:
a = []
with open("csv.txt") as myfile:
firstline = True
for line in myfile:
if firstline:
mykeys = "".join(line.split()).split(',')
firstline = False
else:
values = "".join(line.split()).split(',')
a.append({mykeys[n]:values[n] for n in range(0,len(mykeys))})
# similar solution via namedtuple:
import csv
from collections import namedtuple
with open('foo.csv') as f:
fh = csv.reader(open(f, "rU"), delimiter=',', dialect=csv.excel_tab)
headers = fh.next()
Row = namedtuple('Row', headers)
list_of_dicts = [Row._make(i)._asdict() for i in fh]
Простой метод анализа CSV в список словарей
with open('/home/mitul/Desktop/OPENEBS/test.csv', 'rb') as infile:
header = infile.readline().split(",")
for line in infile:
fields = line.split(",")
entry = {}
for i,value in enumerate(fields):
entry[header[i].strip()] = value.strip()
data.append(entry)
Чтобы преобразовать CSV файл (два столбца, несколько строк, без заголовка) в список словарей, я использовал модуль csv. Мой файл csv выглядел так:
c1,--------
c14,EAE23ED3
c15,--------
Я хотел создать список словарей, каждая строка файла csv будет словарем (ключ, пара значений). Выход, который я хотел:
[
{
"c1": "--------"
},
{
"c14": "EAE23ED3"
},
{
"c15": "--------"
}
]
Для этого я использовал следующий код:
import csv
csv_path = '.../input_file.csv'
mylist = []
mydict = {}
# read the csv and write to a dictionary
with open(csv_path, 'rb') as csv_file:
reader = csv.reader(csv_file)
for row in reader:
mydict = {row[0]:row[1]}
mylist.append(mydict)
print(mylist)
Это работает в моем случае. Чтобы решить мою проблему, эти сообщения были полезны: