Question

对于以下CSV文件：

A,B,C

A1,B1,C1
A1,B2,C2
A2,B3,C3
A2,B4,C4

如何让我的字典看起来像这样：

{'A1': {'B1': 'C1', 'B2' : 'C2'}, 'A2': {'B3': 'C3', 'B4' : 'C4'}}

我使用以下代码： -

EgDict = {}
with open('foo.csv') as f:
    reader = csv.DictReader(f)
    for row in reader:
        key = row.pop('A')
        if '-' in key:
                continue
        if key not in result:
                new_row = {row.pop('B'): row.pop('C')}
                result[key] = new_row
        else:
                result[key][row.pop('B')].append(row.pop('C'))

我收到以下错误： -

AttributeError: 'dict' object has no attribute 'append'

我在这里做错了什么？

Answer 1

字典没有append方法。这将是列表。

您可以考虑使用以下代码：

d = {}
with open('foo.csv') as f:
    reader = csv.DictReader(f)
    for row in reader:
        d.setdefault(row['A'], {}).update({row['B']: row['C']})
print(d)
# {'A1': {'B1': 'C1', 'B2' : 'C2'}, 'A2': {'B3': 'C3', 'B4' : 'C4'}}

setdefault设置一个具有默认{}值的新密钥，当一个密钥不存在且更新方法使用新值更新该密钥时的字典。

Answer 2

res = {}
with open('foo.csv') as f:
     reader = csv.reader(f)
     for i,item in enumerate(reader):
         if i == 0: continue 
         a,b = item[0],item[1:]
         if a not in res:
            res[a] = {}
         res[a].update( {b[0]:b[1]} )

从CSV文件创建字典，

2 个答案: