元素的所有成对乘法,不包括相同元素

问题描述

我有一个数字字典,我想获得所有成对乘法的列表,但不包括自身乘法。

test_dict = {'id_1': 8,'id_2': 9,'id_3': 10}
test_keys = list(test_dict.keys())
list_of_multiples = []

for i in range(0,len(test_keys)):
    count = 0
    
    if i == count:
        count += 1
        
    else:
        if count < len(test_keys):
        
            mult = (test_dict[test_keys[i]] * test_dict[test_keys[count]])
            list_of_multiples.append(mult)
            count += 1

list_of_multiples

输出:[72,80]

预期输出:[72,80,72,90,80,90]

我相对较新,因此if逻辑一直令人困惑。同样,如果有一个预建函数可以做到这一点,那也很好。非常感谢

解决方法

您要查找的内容是使用列表理解功能通过最惯用的python实现的:

keys=test_dict.keys()
list_of_multiples=[test_dict[i]*test_dict[j] for i in keys for j in keys if i!=j]

如果您是python新手,这不一定很明显。这是一种编写以下内容的紧凑方式:

keys=test_dict.keys()
list_of_multiples=[]
for i in keys:
  for j in keys:
    if i != j:
      list_of_multiples.append(test_dict[i]*test_dict[j])
,

使用itertools.permutations

from itertools import permutations

test_dict = {'id_1': 8,'id_2': 9,'id_3': 10}

perm = list(permutations(test_dict.values(),2))
list_of_multiples = [i[0]*i[1] for i in perm]
print(list_of_multiples)

输出:

[72,80,72,90,90]