问题描述
我有一个数字字典,我想获得所有成对乘法的列表,但不包括自身乘法。
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]