迭代 O(n^2 / 2) 一个字典

给定字典:


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])

我怎么能用字典做类似的事情?


慕码人8056858
浏览 183回答 3
3回答

哆啦的时光机

Python 附带电池,但最干净的方法并不总是显而易见的。您已经拥有想要内置的功能itertools。试试这个:import itertoolsresult = {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}

一只斗牛犬

您可以为此使用 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&nbsp;&nbsp; &nbsp; &nbsp; for i,(k,v) in enumerate(d.items())&nbsp;&nbsp; &nbsp; &nbsp; for j,(l,w) in enumerate(d.items())&nbsp;&nbsp; &nbsp; &nbsp; if i<=j}>>> {'b#b': 0, 'b#a': 0, 'b#c': 0, 'a#a': 1, 'a#c': 2, 'c#c': 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&nbsp;[('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}
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python