python - 在 O(n^2/2) 中迭代字典

Question

给定一个字典:

d = {'a':0, 'b': 1, 'c': 2}

我想制作一个新字典来计算 d 和 d 值的乘积。

这是我需要的结果:

d = {'a#a': 0, 'a#b': 0, 'a#c': 0, 'b#b' : 1, 'b#c': 2, 'c#c': 4}

但是我不想得到这样的结果:

d = {'a#a': 0, 'a#b': 0, 'a#c': 0, 'b#a' : 0, 'b#b' : 1, 'b#c': 2, 'c#a': 0, 'c#b': 2, 'c#c': 4}

因为 c#a 已经被 a#c 计算过了。

如果这是一个数组或列表，我会做类似的事情

res = []
t = [0, 1, 2]

for i in range(len(t):
    for j in range(i):
        res.append(t[i] * t[j])

我怎样才能做类似但使用字典的事情？

Answer 1

Python 自带电池，但最干净的方法并不总是显而易见的。您已经在 itertools 中内置了您想要的功能。

试试这个:

import itertools
result = {f'{k1}#{k2}': d[k1]*d[k2]
   for k1, k2 in itertools.combinations_with_replacement(d, 2)}

itertools.combinations 为您提供所有没有重复的对，itertools.combinations_with_replacement 为您提供唯一的对，包括键相同的对。

输出:

>>> print(result)
{'a#a': 0, 'a#b': 0, 'a#c': 0, 'b#b': 1, 'b#c': 2, 'c#c': 4}