从.csv文件创建数据库

时间:2015-07-09 15:34:07

标签: python csv

我需要一个.csv文件,打开文件,读取文件,然后创建一个数据库,这是一个字典,其键是年,值是一个带有元组的列表,如下所示:

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

我需要的类别是“姓名”,类别,死亡,DamageMillions

这是我正在使用的数据的快照:

enter image description here

我该怎么做?

编辑:我知道如何打开和阅读文件,我只是不知道如何创建数据库。

2 个答案:

答案 0 :(得分:2)

这是未经测试的,但应该会让您知道从哪里开始:

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

有关defaultdictcsv模块的详细信息。

答案 1 :(得分:0)

将csv模块与DictReader结合使用。 您将获得一个要迭代的词典列表,您必须进行转换以获得所需的年份索引。

请参阅:https://docs.python.org/2/library/csv.html