我如何在dict理解中将每n个元素设置为dict值

时间:2020-03-31 20:54:13

标签: python dictionary dictionary-comprehension

一个例子:

data_1 = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36]
data_2 = ['key1','key2','key3','key4','key5','key6','key7','key8','key9','key10','key11','key12']
data_3 = {'123':{'a':1},'222':{'b':2}, '333':{'b':3}}

我需要设置此列表以按下一个条件来决定键,data_1中的每个12个元素应该是data_2中每个键的值,而这个dict应该是数据3中键的值。

预期输出:

{'123':{'key1':1,'key2':2,'key3':3,'key4':4,'key5':5,'key6':6,'key7':7,'key8':8,'key9':9,'key10':10,'key11':11,'key12':12},
{'222':{'key1':13,'key2':14,'key3':15,'key4':16,'key5':17,'key6':18,'key7':19,'key8':20,'key9':21,'key10':22,'key11':23,'key12':24},
{'333':{'key1':25,'key2':26,'key3':27,'key4':28,'key5':29,'key6':30,'key7':31,'key8':32,'key9':33,'key10':34,'key11':35,'key12':36}

我的尝试:

    result = {
        key: dict(zip(data_2,data_1) for key, value in data_3.items()
    }

我的结果:

{'123': {'key1': 1, 'key2': 2, 'key3': 3, 'key4': 4, 'key5': 5, 'key6': 6, 'key7': 7, 'key8': 8, 'key9': 9, 'key10': 10, 'key11': 11, 'key12': 12}, '222': {'key1': 1, 'key2': 2, 'key3': 3, 'key4': 4, 'key5': 5, 'key6': 6, 'key7': 7, 'key8': 8, 'key9': 9, 'key10': 10, 'key11': 11, 'key12': 12}, '333': {'key1': 1, 'key2': 2, 'key3': 3, 'key4': 4, 'key5': 5, 'key6': 6, 'key7': 7, 'key8': 8, 'key9': 9, 'key10': 10, 'key11': 11, 'key12': 12}}

我了解这部分陈述中的错误:

键:dict(zip(data_2,data_1)

我如何达到预期的结果?

1 个答案:

答案 0 :(得分:4)

一种方法是利用zip仅在消耗较小的可迭代对象之前进行迭代的事实,因此您可以执行以下操作:

iterable_data1 = iter(data_1)

result = {key: dict(zip(data_2, iterable_data1)) for key in data_3}

for key, value in result.items():
    print(key, value)

这非常有效,因为它不会创建任何其他列表。另外,您可以执行以下操作:

result = {key: dict(zip(data_2, data_1[12 * i:])) for i, key in enumerate(data_3)}

for key, value in result.items():
    print(key, value)

输出

123 {'key1': 1, 'key2': 2, 'key3': 3, 'key4': 4, 'key5': 5, 'key6': 6, 'key7': 7, 'key8': 8, 'key9': 9, 'key10': 10, 'key11': 11, 'key12': 12}
222 {'key1': 13, 'key2': 14, 'key3': 15, 'key4': 16, 'key5': 17, 'key6': 18, 'key7': 19, 'key8': 20, 'key9': 21, 'key10': 22, 'key11': 23, 'key12': 24}
333 {'key1': 25, 'key2': 26, 'key3': 27, 'key4': 28, 'key5': 29, 'key6': 30, 'key7': 31, 'key8': 32, 'key9': 33, 'key10': 34, 'key11': 35, 'key12': 36}

第二种方法的想法是让key中的每个data_3将指针移动data_1 12个位置。

相关问题