Python CSV在读入字典后搜索特定值

Question

我需要一些帮助，使用 Python 将特定值读取到字典中。我有一个包含用户编号的 csv 文件。所以用户 1,2,3...每个用户都在特定的部门 1,2,3...并且每个部门都在特定的建筑物 1,2,3...所以我需要知道如何列出1 号楼的部门 1 的所有用户，然后是 1 号楼的部门 2 的所有用户，依此类推。我一直在尝试并使用 csv.ReadDict 将所有内容读入大型字典中，但是如果我可以搜索我读入每个字典字典中的条目，那么这将起作用。关于如何排序这个文件有什么想法吗？ CSV 包含超过 150,000 个用户条目。每行是一个新用户，列出 3 个属性，user_name、departmentnumber、department Building。有100个部门、100栋大楼和15万用户。有什么想法可以用一个简短的脚本来解决所有这些问题吗？感谢您提前的帮助

Answer 1

蛮力方法看起来

import csv
csvFile = csv.reader(open('myfile.csv'))
data = list(csvFile)
data.sort(key=lambda x: (x[2], x[1], x[0]))

可以扩展到

import csv
import collections

csvFile = csv.reader(open('myfile.csv'))
data = collections.defaultdict(lambda: collections.defaultdict(list))

for name, dept, building in csvFile:
    data[building][dept].append(name)

buildings = data.keys()
buildings.sort()
for building in buildings:
    print "Building {0}".format(building)

    depts = data[building].keys()
    depts.sort()
    for dept in depts:
        print "  Dept {0}".format(dept)

        names = data[building][dept]
        names.sort()
        for name in names:
            print "   ",name