Chuyển đổi nội dung của một tập tin CSV vào một cuốn từ điển

Question

Code tôi có cho đến nay là một chức năng cơ bản đọc một tập tin csv và in nội dung của nó:

def read(filename): 
    with open(filename, 'r') as csvfile: 
     reader = csv.reader(csvfile, delimiter=',') 
     for row in reader: 
      print(row) 

Nội dung sailor.csv:

name, mean performance , std dev 
Alice, 100, 0, 
Bob, 100, 5, 
Clare, 100, 10, 
Dennis, 90, 0, 
Eva, 90, 5, 

read('sailor.csv') và chạy chức năng

sản lượng hiện tại:

['name', ' mean performance ', ' std dev'] 
['Alice', ' 100', ' 0', ''] 
['Bob', ' 100', ' 5', ''] 
['Clare', ' 100', ' 10', ''] 
['Dennis', ' 90', ' 0', ''] 
['Eva', ' 90', ' 5', ''] 

cần đầu ra:

{'Dennis': (90.0, 0.0), 'Clare':(100.0, 10.0), 
'Eva': (90.0, 5.0), 'Bob': (100.0, 5.0), 'Alice': (100.0, 0.0)} 

bất kỳ ý tưởng làm thế nào tôi có thể đạt được kết quả đó? Sử dụng Python 3.4.2 nếu điều đó giúp, giải thích về câu trả lời của bạn sẽ được đánh giá cao!

Answer 1

Tôi nghĩ rằng đây là những gì bạn muốn:

import csv 

def read(filename): 
    out_dict = {} 
    with open(filename, 'r') as csvfile: 
     reader = csv.reader(csvfile, delimiter=',') 
     next(csvfile) # skip the first row 
     for row in reader: 
      out_dict[row[0]] = float(row[1]), float(row[2]) 
      print(row) 

    return out_dict 

print(read('data.csv')) 

Prints:

{'Bob': (' 100', ' 5'), 'Clare': (' 100', ' 10'), 'Alice': (' 100', ' 0'), 'Dennis': (' 90', ' 0'), 'Eva': (' 90', ' 5')} 

Không nhiều để giải thích ở đây. Chỉ cần đặt các giá trị trong từ điển và bỏ qua hàng đầu tiên được thêm vào. Tôi cho rằng tên người là duy nhất.

Answer 2

sử dụng thư viện chuẩn csv và một sự hiểu biết từ điển ...

import csv 
with open('sailor.csv') as csvfile: 
    reader = csv.reader(csvfile) 
    next(reader) 
    d = {r[0] : tuple(r[1:-1]) for r in reader} 

đâu d sẽ là từ điển mà bạn muốn. d[1:-1] cắt mảng từ thứ hai sang phần tử thứ hai thành phần tử cuối cùng.

EDIT: bỏ qua tiêu đề, chuyển đến các bộ

Answer 3

Sử dụng DictReader:

def read(filename): 
    with open(filename, 'r') as csvfile: 
     reader = csv.DictReader(csvfile, delimiter=',') 
     for row in reader: 
      print(row) 

Answer 4

Vì vậy ... Tôi biết câu hỏi này đã được trả lời chủ yếu, nhưng tôi nghĩ rằng tôi chỉ muốn ném một lớp lót trong hỗn hợp để thêm vào để rút ngắn câu trả lời:

from csv import reader 
from itertools import islice 

{r[0] : tuple(r[1:-1]) for r in islice(reader(open('sailor.csv')), 1, None)} 

điều duy nhất thực sự cuốn tiểu thuyết được thêm islice bỏ qua dòng tiêu đề sạch.

Answer 5

Đây là giải pháp của tôi nếu tôi có thể:

>>> import pyexcel as pe 
>>> s = pe.load("sailor.csv", name_rows_by_column=0, name_columns_by_row=0) 
>>> s.format(float) 
>>> s 
Sheet Name: csv 
+--------+------------------+---------+---+ 
|  | mean performance | std dev | | 
+========+==================+=========+===+ 
| Alice | 100    | 0  | 0 | 
+--------+------------------+---------+---+ 
| Bob | 100    | 5  | 0 | 
+--------+------------------+---------+---+ 
| Clare | 100    | 10  | 0 | 
+--------+------------------+---------+---+ 
| Dennis | 90    | 0  | 0 | 
+--------+------------------+---------+---+ 
| Eva | 90    | 5  | 0 | 
+--------+------------------+---------+---+ 
>>> del s.column[''] # delete the column which has '' as its name 
>>> s.to_dict(True) # make a dictionary using row names as key 
OrderedDict([('Alice', [100.0, 0.0]), ('Bob', [100.0, 5.0]), 
('Clare', [100.0, 10.0]), ('Dennis', [90.0, 0.0]), ('Eva', [90.0, 5.0])]) 

Dưới đây là tài liệu trên pe.load và to_dict của pyexcel