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}

Answer 2

您可以为此使用dict理解：

dd = {f'{k}#{l}': v*w for k,v in d.items() for l,w in d.items() if k<=l}
>>> {'a#a': 0, 'a#b': 0, 'a#c': 0, 'b#b': 1, 'b#c': 2, 'c#c': 4}

编辑：如果您希望将结果按d中的物品外观排序：

d = {'b': 0, 'a': 1, 'c': 2}
dd = {f'{k}#{l}': v*w 
      for i,(k,v) in enumerate(d.items()) 
      for j,(l,w) in enumerate(d.items()) 
      if i<=j}
>>> {'b#b': 0, 'b#a': 0, 'b#c': 0, 'a#a': 1, 'a#c': 2, 'c#c': 4}

Answer 3

为什么不使用dictionary comprehension？

{f"{k1}#{k2}": d[k1] * d[k2] for k1 in d for k2 in d if k1 <= k2}

或者，如果您喜欢传统风格

result = {}
for k1 in d:
    for k2 in (k for k in d if k >= k1):
        result[f"{k1}#{k2}"] = d[k1] * d[k2]

后一种形式使用生成器，因此不会每次都创建单独的键列表。

注意：如果键不具有可比性（例如，不是整数，字符串等），则将不起作用，因此您应使用if hash(k1) < hash(k2)（它适用于所有类型的键，因为字典键必须是可散列的，但键的顺序可能不直观。

Answer 4

您可以使用itertools获得组合并形成字典！

>>> from itertools import combinations
>>>
>>> d
{'a': 0, 'c': 2, 'b': 1}
>>> combinations(d.keys(),2) # this returns an iterator
<itertools.combinations object at 0x1065dc100>
>>> list(combinations(d.keys(),2)) # on converting them to a list 
[('a', 'c'), ('a', 'b'), ('c', 'b')]
>>> {"{}#{}".format(v1,v2): (v1,v2) for v1,v2 in combinations(d.keys(),2)} # form a dict using dict comprehension, with "a#a" as key and a tuple of two values.
{'a#c': ('a', 'c'), 'a#b': ('a', 'b'), 'c#b': ('c', 'b')}
>>> {"{}#{}".format(v1,v2): d[v1]*d[v2] for v1,v2 in combinations(d.keys(),2)}
{'a#c': 0, 'a#b': 0, 'c#b': 2} # form the actual dict with product as values
>>> {"{}#{}".format(v1,v2):d[v1]*d[v2] for v1,v2 in list(combinations(d.keys(),2)) + [(v1,v1) for v1 in d.keys()]} # form the dict including the self products!
{'a#c': 0, 'a#b': 0, 'a#a': 0, 'b#b': 1, 'c#c': 4, 'c#b': 2}

或者像邓肯所指出的那样简单

>>> from itertools import combinations_with_replacement
>>> {"{}#{}".format(v1,v2): d[v1]*d[v2] for v1,v2 in combinations_with_replacement(d.keys(),2)}
{'a#c': 0, 'a#b': 0, 'a#a': 0, 'b#b': 1, 'c#c': 4, 'c#b': 2}

Answer 5

itertools有趣：

from itertools import combinations_with_replacement

letter_pairs, number_pairs = (combinations_with_replacement(l, r=2) for l in zip(*d.items()))
result = {letters: number 
          for letters, number in zip(map('#'.join, letter_pairs), 
                                     (a * b for a, b in number_pairs))}

print(result)

输出：

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

在O（n ^ 2/2）中迭代一个字典

5 个答案: