從.csv文件創建數據庫

Question

我需要獲取一個.csv文件，打開該文件，讀取該文件，然后創建一個數據庫，這是一本字典，其鍵是年，值是一個包含元組的列表，如下所示：

db = {2002:[('Andrew', 5, 63, 75000)], etc…}

我需要的類別是“名稱”，類別，死亡，損害百萬

這是我正在使用的數據的快照：

我該怎么做？

編輯：我知道如何打開和讀取文件，我只是不知道如何創建數據庫。

Answer 1

這未經測試，但是應該給您一個好主意，從哪里開始：

import csv
from collections import defaultdict

db = defaultdict(list)
with open('csvfile.csv') as f:
     csvreader = csv.reader(f)
     for row in csvreader:
         db[row[0]].append((row[1], row[4], row[5], row[6]))

return db

有關defaultdict和csv模塊的更多信息。

Answer 2

將csv模塊與DictReader結合使用。 您將獲得要迭代的字典列表，必須對其進行轉換才能獲得所需的年份索引。

參見： https : //docs.python.org/2/library/csv.html