Tạo một từ điển từ một tập tin csv?

Tôi đang cố gắng tạo một từ điển từ tệp csv. Cột đầu tiên của tệp csv chứa các khóa duy nhất và cột thứ hai chứa các giá trị. Mỗi hàng của tệp csv đại diện cho một cặp khóa, giá trị duy nhất trong từ điển. Tôi đã cố gắng sử dụng csv.DictReadervà csv.DictWritercác lớp, nhưng tôi chỉ có thể tìm ra cách tạo một từ điển mới cho mỗi hàng. Tôi muốn một từ điển. Đây là mã tôi đang cố gắng sử dụng:

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)
    for rows in reader:
        k = rows[0]
        v = rows[1]
        mydict = {k:v for k, v in rows}
    print(mydict)

Khi tôi chạy đoạn mã trên tôi nhận được a ValueError: too many values to unpack (expected 2). Làm cách nào để tạo một từ điển từ tệp csv? Cảm ơn.

Tôi tin rằng cú pháp bạn đang tìm kiếm như sau:

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}

Cách khác, đối với python <= 2.7.1, bạn muốn:

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

Mở tệp bằng cách gọi mở và sau đó csv.DictReader.

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

Bạn có thể lặp qua các hàng của đối tượng đọc tệp csv bằng cách lặp qua input_file.

for row in input_file:
    print(row)

HOẶC Chỉ truy cập dòng đầu tiên

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

CẬP NHẬT Trong các phiên bản python 3+, mã này sẽ thay đổi một chút:

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

import csv
reader = csv.reader(open('filename.csv', 'r'))
d = {}
for row in reader:
   k, v = row
   d[k] = v

Đây không phải là thanh lịch mà là một giải pháp một dòng sử dụng gấu trúc.

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

Nếu bạn muốn chỉ định dtype cho chỉ mục của mình (nó không thể được chỉ định trong read_csv nếu bạn sử dụng đối số index_col vì lỗi ):

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

Bạn chỉ cần chuyển đổi csv.reader sang 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'}

Bạn cũng có thể sử dụng numpy cho việc này.

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

Tôi khuyên bạn nên thêm if rowsvào trong trường hợp có một dòng trống ở cuối tệp

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)

Giải pháp một lớp lót

import pandas as pd

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

Nếu bạn ổn với việc sử dụng gói numpy, thì bạn có thể làm một cái gì đó như sau:

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]

Đối với các tệp csv đơn giản, chẳng hạn như sau

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

Bạn có thể chuyển đổi nó thành một từ điển Python chỉ bằng cách sử dụng

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)}

Điều này sẽ mang lại từ điển sau

{'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'}}

Lưu ý: Từ điển Python có các khóa duy nhất, vì vậy nếu tệp csv của bạn trùng lặp, idsbạn nên nối từng hàng vào danh sách.

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)})

Bạn có thể sử dụng cái này, nó khá tuyệt:

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

Nhiều giải pháp đã được đăng và tôi muốn đóng góp cho tôi, giải pháp này hoạt động cho một số cột khác nhau trong tệp CSV. Nó tạo ra một từ điển với một khóa cho mỗi cột và giá trị cho mỗi khóa là một danh sách với các thành phần trong cột đó.

    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])

với gấu trúc, nó dễ dàng hơn nhiều, ví dụ. giả sử bạn có dữ liệu sau dưới dạng CSV và hãy gọi nó test.txt/ test.csv(bạn biết CSV là một loại tệp văn bản)

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

hiện đang sử dụng gấu trúc

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

cho mỗi hàng, nó sẽ là

df.to_dict(orient='records')

và đó là nó.

Hãy thử sử dụng một defaultdictvà 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)

Nó trở lại:

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