使用 for 循环在字典中创建内部键

Question

我正在使用 for 循环从表中创建字典。目标是使用 for 循环创建具有嵌套键值对的字典。

表格为文本格式，第一行是表头，如下所示：

AAA|BBB|CCC
zzz|100|xy
zzz|50|xy
xxx|200|xy
xxx|100|xy
zzz|300|xz
zzz|400|xz
xxx|200|xz
xxx|700|xz

要按 AAA 获取总数，（在 AAA:CCC 键值对中），代码成功：

dict = {}

with open('data.txt') as table:
    table.readline()
    for line in table:
        (AAA, BBB, CCC) = line.split("|")
        dict[AAA] = dict[AAA] + int(BBB) if AAA in dict else int(BBB)

输出如预期：

{'zzz': 850, 'xxx': 1200}

下一步（也是我卡住的地方）是如何执行迭代以创建内部键，格式为

{'zzz':{xy:150, xz:700}, 'xxx':{xy:300, xz:900}}

这是我目前所拥有的，但不明白如何将 insideKey 插入到这个字典中：

dict = {}
new_dict = {}

with open('data.txt') as table:
    table.readline()
    for line in table:
        (AAA, BBB, CCC) = line.split("|")
        dict[AAA] = dict[AAA] + (new_dict[CCC] + int(BBB)) if CCC in new_dict else int(BBB)

我了解我无法使用 + 添加 insideKey，但找不到执行此操作的方法。

预期输出：

{'zzz':{xy:150, xz:700}, 'xxx':{xy:300, xz:900}}

Answer 1

您可以使用setdefault() 设置初始值。循环时只需设置适当的默认值或使用get 获取内部值或零：

d = {}
with open('data.txt') as table:
    next(table)
    for line in table:
        (AAA, BBB, CCC) = map(str.strip, line.split("|"))
        outer = d.setdefault(AAA, {})
        outer[CCC] = outer.get(CCC, 0) + int(BBB)

结果：

{'zzz': {'xy': 150, 'xz': 700}, 'xxx': {'xy': 300, 'xz': 900}}

Answer 2

一个快速的建议：不要使用 'dict' 作为字典 instance 的名称，因为它也是字典 class 的名称。 Python 允许您将绑定修改为内置名称。这很强大，但是如果你在那之后需要使用 dict() 构造函数，你将无法做到。

现在，谈谈你的问题。我认为您试图在太少的代码行中完成太多工作。编写多行明确的行来代替该字典分配并不可耻。我还建议您通过构建更简单的字典作为中间步骤来重新组织问题。

字典键可以是任何可以散列的不可变数据类型。这包括仅包含不可变数据的元组。该程序以您想要的方式为您提供汇总的数据，但并没有完全按照您想要的方式组织：

dct = {}
with open('data.txt') as table:
    table.readline()
    for line in table:
        (AAA, BBB, CCC) = line.split("|")
        v = int(BBB)     # value
        k = (AAA, CCC)   # key
        try:
            dct[k] += v  # existing key
        except KeyError:
            dct[k] = v   # new key

此时，dct包含：

{('zzz', 'xy'): 150, ('xxx', 'xy'): 300, ('zzz', 'xz'): 700, ('xxx', 'xz'): 900}

现在，使用第二个循环创建一个新字典，按照您想要的方式重新组织。

result = {}
for (k1, k2), v in dct.items():
    try:
        result[k1][k2] = v
    except KeyError:
        result[k1] = {k2 : v}

这是结果中的内容：

{'zzz': {'xy': 150, 'xz': 700}, 'xxx': {'xy': 300, 'xz': 900}}

我使用了 Python 初学者可能不知道的三个技巧：比字符串或数字更复杂的字典键；异常处理;和元组拆包。如果您还没有学习这些概念，您可能会发现它们非常有用。

注意：如果您可以使用第三方包，Pandas DataFrames 是完成此类工作的绝佳工具。不过，如果您有大量数据处理工作要做，Pandas 可能才值得付出努力。

Answer 3

只是为了说明使用 defaultdict。

from collections import defaultdict

d = defaultdict(dict)

with open('f0714.txt', 'r') as table:
    next(table)
    for line in table:
        (AAA, BBB, CCC) = map(str.strip, line.split("|"))

        if CCC not in d[AAA]:
            d[AAA][CCC] = int(BBB)
        else:
            d[AAA][CCC] += int(BBB)

print(dict(d))

'''
{'zzz': {'xy': 150, 'xz': 700}, 'xxx': {'xy': 300, 'xz': 900}}
'''