如何从一个文件创建twoleveldictionary？

Question

我有一个csv文件，看起来像这样（实际上它更大）：

country;company1;company2;company3
finland;30;30;40
sweden;20;30;50
norway;10;20;70

我怎样才能最容易地读取此文件，以便获得这样的词典（词典中的词典）：

{ 'company1': {'finland': 30, 'sweden': 20, 'norway': 10}
'company2': {'finland': 30, 'sweden': 30, 'norway': 20}
... 
}

我首先试着从文件的第一个单独列表，也就是从公司，然后从他们创建了一个字典。但是，当我试图阅读第一行之后的行并在已经创建的行中创建字典时，遇到了问题

如果解释不好，我很抱歉，我对编码是新手

Answer 1

@fsimonjetz的answer非常棒，如果你已经在这个项目中与熊猫合作。如果您没有这样做，那么仅将其用于此任务是一种巨大的过度使用，因为我们可以用简单的逻辑解析和转换数据

import csv

from collections import defaultdict

output = defaultdict(dict)

with open('path/to/your.csv') as f:
    reader = csv.DictReader(f, delimiter=';')
    companies = reader.fieldnames[1:]
    for line in reader:
        country = line['country']
        for company in companies:
            output[company][country] = line[company]
            # or output.setdefault(company, {})[country] = line[company]
            # if you want 'output' to be a "normal" dict instead of defaultdict

print(dict(output))  # or just print(output) if you don't mind seeing OrderedDict
                     # repr

输出

{'company1': {'finland': '30', 'sweden': '20', 'norway': '10'}, 
 'company2': {'finland': '30', 'sweden': '30', 'norway': '20'}, 
 'company3': {'finland': '40', 'sweden': '50', 'norway': '70'}}