我有2个示例列表,我想要实现的是获得一个嵌套默认字典,其中包含值的总和。
以下代码很好用:
from collections import defaultdict
l1 = [1,2,3,4]
l2 = [5,6,7,8]
dd = defaultdict(int)
for i in l1:
for ii in l2:
dd[i] += ii
但是我想做的是在d
字典中创建一个默认密钥:
from collections import defaultdict
l1 = [1,2,3,4]
l2 = [5,6,7,8]
dd = defaultdict(int)
for i in l1:
for ii in l2:
dd[i]['mykey'] += ii
这会引发错误:
Traceback (most recent call last):
File "/usr/lib/python3.6/code.py", line 91, in runcode
exec(code, self.locals)
File "<input>", line 1, in <module>
File "<string>", line 12, in <module>
TypeError: 'int' object is not subscriptable
基本上我无法理解的是,是否有机会将defaultdict(dict)
和defaultdict(int)
混合使用。
答案 0 :(得分:5)
您想要默认整数的默认字典:
dd = defaultdict(lambda: defaultdict(int))
运行代码后:
>>> dd
{1: defaultdict(<class 'int'>, {'mykey': 26}),
2: defaultdict(<class 'int'>, {'mykey': 26}),
3: defaultdict(<class 'int'>, {'mykey': 26}),
4: defaultdict(<class 'int'>, {'mykey': 26})}
答案 1 :(得分:5)
defaultdict数据结构接收一个将提供默认值的函数,因此,如果要创建一个defautdict(int)
作为默认值,请提供一个执行该功能的函数,例如lambda : defaultdict(int)
:
from collections import defaultdict
from pprint import pprint
l1 = [1, 2, 3, 4]
l2 = [5, 6, 7, 8]
dd = defaultdict(lambda : defaultdict(int))
for i in l1:
for ii in l2:
dd[i]['mykey'] += ii
pprint(dd)
输出
defaultdict(<function <lambda> at 0x7efc74d78f28>,
{1: defaultdict(<class 'int'>, {'mykey': 26}),
2: defaultdict(<class 'int'>, {'mykey': 26}),
3: defaultdict(<class 'int'>, {'mykey': 26}),
4: defaultdict(<class 'int'>, {'mykey': 26})})