[英]Building a dictionary with lists as values using list comprehension
我正在尝试使用list/dict comprehension来简化以下代码,其中我想从列表中构建列表字典。
在此示例中,我将每个项目的 mod(或任何其他功能)作为键。 如果字典中不存在该键,则创建一个新列表并将该项目附加到列表中,否则直接附加该项目。
def get_key(val):
return val % 3 # Can be something else, it doesn't matter
dict = {}
data = [i for i in range(10)]
for item in data:
key = get_key(item)
if key in dict:
dict[key].append(item)
else:
dict[key] = [item]
print(dict)
预期 output
{0: [0, 3, 6, 9], 1: [1, 4, 7], 2: [2, 5, 8]}
为了简化上面的代码,我尝试了这种丑陋的方法:
for item in data:
key = get_key(item)
dict[key] = [item] if key not in dict else (dict[key], dict[key].append(item))[0]
但是,如何在没有 for 循环的情况下使用list/dict 理解作为单个表达式来实现相同的目标?
我可以想到这样的事情,但它无法将 append 值放入预先存在的列表中。
{get_key(item):[item] for item in data}
相关文章:
你很亲密。 下面是我使用字典理解的方法
data = list(range(0,10))
mod= 3
{i:data[i::mod] for i in range(mod)}
出去
{0: [0, 3, 6, 9], 1: [1, 4, 7], 2: [2, 5, 8]}
创建一个值列表,每个子列表的第一个元素是键,即[n] +
,然后使用它来创建字典。
m = [[n] + [i[1] for i in [(x%3,x) for x in range(10)]if i[0] == n] for n in range(3)]
dictt = {i[0]:i[1:] for i in m}
print(dictt)
>>> {0: [0, 3, 6, 9], 1: [1, 4, 7], 2: [2, 5, 8]}
全部在一条线上
dictt = {i[0]:i[1:] for i in [[n] + [i[1] for i in [(x%3,x) for x in range(10)]if i[0] == n] for n in range(3)]}
您在这里并不需要列表/字典理解。 只需使用带有defaultdict
的普通循环:
out = collections.defaultdict(list)
for i in range(10):
out[i % 3].append(i)
尝试默认 dict ,
from collections import defaultdict
d = defaultdict(list)
for item in range(10):
d[item % 3].append(item)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.