Создание словаря из файла csv?

Я считаю, что синтаксис, который вы искали, выглядит следующим образом:

import csv

with open('coors.csv', mode='r') as infile:
    reader = csv.reader(infile)
    with open('coors_new.csv', mode='w') as outfile:
        writer = csv.writer(outfile)
        mydict = {rows[0]:rows[1] for rows in reader}

В качестве альтернативы, для python ‹= 2.7.1 вы хотите:

mydict = dict((rows[0],rows[1]) for rows in reader)

Откройте файл, вызвав open, а затем используя csv.DictReader.

input_file = csv.DictReader(open("coors.csv"))

Вы можете перебирать строки объекта чтения dict файла csv, перебирая input_file.

for row in input_file:
    print(row)

ИЛИ Для доступа только к первой строке

dictobj = csv.DictReader(open('coors.csv')).next() 

ОБНОВЛЕНИЕ В версиях Python 3+ этот код немного изменится:

reader = csv.DictReader(open('coors.csv'))
dictobj = next(reader) 

Это не элегантно, а однострочное решение с использованием панд.

import pandas as pd
pd.read_csv('coors.csv', header=None, index_col=0, squeeze=True).to_dict()

Если вы хотите указать dtype для своего индекса (его нельзя указать в read_csv, если вы используете аргумент index_col из-за ошибка):

import pandas as pd
pd.read_csv('coors.csv', header=None, dtype={0: str}).set_index(0).squeeze().to_dict()

Вам нужно просто преобразовать csv.reader в dict:

~ >> cat > 1.csv
key1, value1
key2, value2
key2, value22
key3, value3

~ >> cat > d.py
import csv
with open('1.csv') as f:
    d = dict(filter(None, csv.reader(f)))

print(d)

~ >> python d.py
{'key3': ' value3', 'key2': ' value22', 'key1': ' value1'}

Вы также можете использовать для этого numpy.

from numpy import loadtxt
key_value = loadtxt("filename.csv", delimiter=",")
mydict = { k:v for k,v in key_value }

Однострочное решение

import pandas as pd

dict = {row[0] : row[1] for _, row in pd.read_csv("file.csv").iterrows()}

Для простых файлов csv, например следующих

id,col1,col2,col3
row1,r1c1,r1c2,r1c3
row2,r2c1,r2c2,r2c3
row3,r3c1,r3c2,r3c3
row4,r4c1,r4c2,r4c3

Вы можете преобразовать его в словарь Python, используя только встроенные модули.

with open(csv_file) as f:
    csv_list = [[val.strip() for val in r.split(",")] for r in f.readlines()]

(_, *header), *data = csv_list
csv_dict = {}
for row in data:
    key, *values = row   
    csv_dict[key] = {key: value for key, value in zip(header, values)}

Это должно дать следующий словарь

{'row1': {'col1': 'r1c1', 'col2': 'r1c2', 'col3': 'r1c3'},
 'row2': {'col1': 'r2c1', 'col2': 'r2c2', 'col3': 'r2c3'},
 'row3': {'col1': 'r3c1', 'col2': 'r3c2', 'col3': 'r3c3'},
 'row4': {'col1': 'r4c1', 'col2': 'r4c2', 'col3': 'r4c3'}}

Примечание. Словари Python имеют уникальные ключи, поэтому, если в вашем CSV-файле есть дубликаты ids, вам следует добавить каждую строку в список.

for row in data:
    key, *values = row

    if key not in csv_dict:
            csv_dict[key] = []

    csv_dict[key].append({key: value for key, value in zip(header, values)})

Я бы предложил добавить if rows на случай, если в конце файла есть пустая строка

import csv
with open('coors.csv', mode='r') as infile:
    reader = csv.reader(infile)
    with open('coors_new.csv', mode='w') as outfile:
        writer = csv.writer(outfile)
        mydict = dict(row[:2] for row in reader if row)

Предполагая, что у вас есть CSV с такой структурой:

"a","b"
1,2
3,4
5,6

И вы хотите, чтобы результат был:

[{'a': '1', ' "b"': '2'}, {'a': '3', ' "b"': '4'}, {'a': '5', ' "b"': '6'}]

Функция zip (пока не упоминается) проста и весьма полезна.

def read_csv(filename):
    with open(filename) as f:
        file_data=csv.reader(f)
        headers=next(file_data)
        return [dict(zip(headers,i)) for i in file_data]

Если вы предпочитаете панд, он также может сделать это довольно хорошо:

import pandas as pd
def read_csv(filename):
    return pd.read_csv(filename).to_dict('records')

Если вы согласны с использованием пакета numpy, вы можете сделать что-то вроде следующего:

import numpy as np

lines = np.genfromtxt("coors.csv", delimiter=",", dtype=None)
my_dict = dict()
for i in range(len(lines)):
   my_dict[lines[i][0]] = lines[i][1]

Вы можете использовать это, это довольно круто:

import dataconverters.commas as commas
filename = 'test.csv'
with open(filename) as f:
      records, metadata = commas.parse(f)
      for row in records:
            print 'this is row in dictionary:'+rowenter code here

с пандами, например, это намного проще. предполагая, что у вас есть следующие данные как CSV, и давайте назовем их test.txt / test.csv (вы знаете, что CSV - это своего рода текстовый файл)

a,b,c,d
1,2,3,4
5,6,7,8

теперь использую панд

import pandas as pd
df = pd.read_csv("./text.txt")
df_to_doct = df.to_dict()

для каждой строки это будет

df.to_dict(orient='records')

вот и все.

Попробуйте использовать defaultdict и DictReader.

import csv
from collections import defaultdict
my_dict = defaultdict(list)

with open('filename.csv', 'r') as csv_file:
    csv_reader = csv.DictReader(csv_file)
    for line in csv_reader:
        for key, value in line.items():
            my_dict[key].append(value)

Он возвращает:

{'key1':[value_1, value_2, value_3], 'key2': [value_a, value_b, value_c], 'Key3':[value_x, Value_y, Value_z]}

Было опубликовано много решений, и я хотел бы внести свой вклад в мой, который работает для другого количества столбцов в файле CSV. Он создает словарь с одним ключом на столбец, а значение для каждого ключа представляет собой список с элементами в таком столбце.

    input_file = csv.DictReader(open(path_to_csv_file))
    csv_dict = {elem: [] for elem in input_file.fieldnames}
    for row in input_file:
        for key in csv_dict.keys():
            csv_dict[key].append(row[key])

Если у вас есть:

1. Только 1 ключ и 1 значение в качестве ключа, значение в вашем CSV
2. Не хочу импортировать другие пакеты
3. Хотите создать диктант одним выстрелом

Сделай это:

mydict = {y[0]: y[1] for y in [x.split(",") for x in open('file.csv').read().split('\n') if x]}

Что оно делает?

Он использует понимание списка для разделения строк и последнее, если x используется для игнорирования пустой строки (обычно в конце), которая затем распаковывается в dict с использованием понимания словаря.