Membuat kamus dari file csv?

Saya mencoba membuat kamus dari file csv. Kolom pertama dari file csv berisi kunci unik dan kolom kedua berisi nilai. Setiap baris file csv mewakili kunci unik, pasangan nilai dalam kamus. Saya mencoba menggunakan csv.DictReaderdan csv.DictWriterkelas, tetapi saya hanya bisa mengetahui cara membuat kamus baru untuk setiap baris. Saya ingin satu kamus. Berikut adalah kode yang saya coba gunakan:

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)

Ketika saya menjalankan kode di atas saya mendapatkan ValueError: too many values to unpack (expected 2). Bagaimana cara membuat satu kamus dari file csv? Terima kasih.

Saya percaya sintaks yang Anda cari adalah sebagai berikut:

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}

Bergantian, untuk python <= 2.7.1, Anda ingin:

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

Buka file dengan memanggil open lalu csv.DictReader.

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

Anda dapat mengulangi baris dari objek pembaca file csv dicter dengan mengulangi input_file.

for row in input_file:
    print(row)

ATAU Untuk mengakses saluran pertama saja

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

PEMBARUAN Dalam versi python 3+, kode ini akan berubah sedikit:

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

Ini tidak elegan tetapi solusi satu baris menggunakan panda.

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

Jika Anda ingin menentukan tipe untuk indeks Anda (itu tidak dapat ditentukan dalam read_csv jika Anda menggunakan argumen index_col karena suatu bug ):

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

Anda harus mengonversi csv.reader menjadi 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'}

Anda juga bisa menggunakan numpy untuk ini.

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

Saya sarankan menambahkan if rowskalau-kalau ada baris kosong di akhir file

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)

Solusi satu lapis

import pandas as pd

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

Jika Anda setuju dengan menggunakan paket numpy, maka Anda dapat melakukan sesuatu seperti berikut:

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]

Untuk file csv sederhana, seperti berikut ini

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

Anda dapat mengonversinya ke kamus Python hanya menggunakan built-in

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

Ini harus menghasilkan kamus berikut

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

Catatan: Kamus python memiliki kunci unik, jadi jika file csv Anda memiliki duplikat idsAnda harus menambahkan setiap baris ke daftar.

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

Anda dapat menggunakan ini, itu sangat keren:

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

Banyak solusi telah diposting dan saya ingin berkontribusi dengan saya, yang berfungsi untuk sejumlah kolom dalam file CSV. Itu menciptakan kamus dengan satu kunci per kolom, dan nilai untuk setiap kunci adalah daftar dengan unsur-unsur dalam kolom tersebut.

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

dengan panda, itu jauh lebih mudah, misalnya. dengan asumsi Anda memiliki data berikut sebagai CSV dan sebut saja test.txt/ test.csv(Anda tahu CSV adalah semacam file teks)

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

sekarang menggunakan panda

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

untuk setiap baris, itu akan menjadi

df.to_dict(orient='records')

dan hanya itu.

Coba gunakan a defaultdictdan 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)

Ia mengembalikan:

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