字典键和值Python的笛卡尔积

问题描述 投票:2回答:1

我有两个数据列表:

COURSES = [C1, C2, C3]
ROOMS = [R1, R2, R3]

我已经创建了一个包含笛卡尔积的元组列表:

L_CR = list(itertools.product(COURSES, ROOMS))
#print(L_CR): [('C1', 'R1'), ('C1', 'R2'), ('C1', 'R3'), ('C2', 'R1').....

现在我创建了两个字典,存储关于参加课程的人数和适合房间的最大人数的整数值:

dic_courses = {'C1': 10, 'C2': 5, 'C3': 20}
dic_rooms = {'R1': 5, 'R2': 10, 'R3': 20}

我无法弄清楚的是我如何创建一个字典,其中元组作为键,以及房间可能达到的最大人数和参加课程的人数之间的绝对差异。到目前为止我所做的是创建一个字典,将每个元组存储为一个键,并为每个键分配一个值1:

Dic_Courses_Room_Capacity = {}
for Element in L_CR:    
    Dic_Courses_Room_Capacity_Temp = {Element: 1}
    Dic_Courses_Room_Capacity.update(Dic_Students_Room_Capacity_Temp)

结果如下:

#print(Dic_Courses_Room_Capacity): {('C1', 'R1'): 1, ('C1', 'R2'): 1, ('C1', 'R3'): 1...

但我要找的是:

#print(Dic_Courses_Room_Capacity): {('C1', 'R1'): 5 , ('C1', 'R2'): 0, ('C1', 'R3'): 10

例如:('C1','R3'):10 = abs('C1' - 'R3')= abs(10-20)= abs(-10)= 10

因此,我必须使用其他两个词典的所有计算组合来更新现有词典(Dic_Courses_Room_Capacity)。我尝试用两个for循环遍历这两个字典,但这导致每个键(=元组)总是得到相同的数字。

有任何想法吗?最好的问候,Jan

python dictionary cartesian-product
1个回答
2
投票

通过两个字典进行简单迭代,为另一个字典添加元组和绝对差异:

dic_courses = {'C1': 10, 'C2': 5, 'C3': 20}
dic_rooms = {'R1': 5, 'R2': 10, 'R3': 20}

d = {}
for k1, v1 in dic_courses.items():
    for k2, v2 in dic_rooms.items():
        d.update({(k1, k2): abs(v1 - v2)})

print(d)

# {('C1', 'R1'): 5, ('C1', 'R2'): 0, ('C1', 'R3'): 10,
#  ('C2', 'R1'): 0, ('C2', 'R2'): 5, ('C2', 'R3'): 15,
#  ('C3', 'R1'): 15, ('C3', 'R2'): 10, ('C3', 'R3'): 0}

或者itertools.product方式:

from itertools import product

dic_courses = {'C1': 10, 'C2': 5, 'C3': 20}
dic_rooms = {'R1': 5, 'R2': 10, 'R3': 20}

d = {}
for x, y in product(dic_courses, dic_rooms):
    d.update({(x, y): abs(dic_courses[x] - dic_rooms[y])})

print(d)

# {('C1', 'R1'): 5, ('C1', 'R2'): 0, ('C1', 'R3'): 10, 
#  ('C2', 'R1'): 0, ('C2', 'R2'): 5, ('C2', 'R3'): 15,
#  ('C3', 'R1'): 15, ('C3', 'R2'): 10, ('C3', 'R3'): 0}
© www.soinside.com 2019 - 2024. All rights reserved.