python - 在 python3 中反转字典中的键和值(值不唯一)

Question

我知道如何在值唯一时简单地反转字典中的键和值。但是当值不唯一时如何反转。根据要求，如果这个值出现不止一次，我需要用set把它们放在一起。

---

例。输入 d = {'a':1, 'b':2,'c':1,'d':2} 输出 d = {1,{'a',' c'},2,{'b','c'}}

---

我在下面写了非常愚蠢的代码，但是因为我只创建了一个集合，所以所有显示超过一次的值都在那个集合中。

def change(d):

    inverted_l = list(map(lambda t:(t[1],t[0]), d.items()))
    store_key = [] #for store the key to check if value appear more than one
    new_d = {}
    x = set()
    for i in range(len(inverted_l)):
        store_key.append(inverted_l[i][0])
    for i in range(len(store_key)):
        if store_key.count(store_key[i])> 1:
            x.add(inverted_l[i][1]) #I think the problem is I need create set
                                    #each time, but I don't know how to do that
            new_d[store_key[i]] = x
        else:
            new_d[store_key[i]] = inverted_l[i][1]
    return new_d

print(sorted(change({'a':1, 'b':2, 'c':1,'d':2}).items()))

我的错误输出是 [(1, {'c', 'd', 'b', 'a'}), (2, {'c', 'd', 'b', ' a'})]但是我需要[(1, {'a', 'c'}), (2, {'b', 'd'})]

补充:我尝试了你的代码，但是当我测试 print(sorted(invert_dict({'a':1, 'b':2, 'c':1}).items()) )我希望我的结果是 [(1, {'a', 'c'}), (2, 'b')]我是 python 的新手，感谢您的帮助和时间!

def invert_dict(d):
    result = {}
    for k in d:
        if d[k] not in result:
            result[d[k]] = set()
        result[d[k]].add(k)
    return {k: d[k] if len(d[k])>1 else d[k].pop() for k in d}



 Traceback (most recent call last):
  File "U:\test.py", line 9, in <module>
    print(sorted(invert_dict({'a':1, 'b':2, 'c':1}).items()))
  File "U:\test.py", line 7, in invert_dict
    return {k: d[k] if len(d[k])>1 else d[k].pop() for k in d}
  File "U:\test.py", line 7, in <dictcomp>
    return {k: d[k] if len(d[k])>1 else d[k].pop() for k in d}
TypeError: object of type 'int' has no len()

Answer 1

我很确定你的意思是所需的输出不是集合

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

而是字典

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

只需仔细检查这里:-)。

无论如何，我会这样做:

import collections

def invert_dict(d):
    result = collections.defaultdict(set)
    for k in d:
        result[d[k]].add(k)
    return dict(result)

return 可以简化为 return result 如果 dict subclass defaultdict 没问题——如果规范对此非常严格，则只需要将其转换为 dict。

我想下一步可​​能是“糟糕，不允许导入”以禁止 collections.defaultdict，所以我预计 - 在这种情况下，改为执行(例如)

def invert_dict(d):
    result = {}
    for k in d:
        if d[k] not in result:
            result[d[k]] = set()
        result[d[k]].add(k)
    return result

添加:显然最新版本是至关重要的(OP 当然“忘记”首先添加“禁止导入”约束——为什么他们一直对我这样做？!他们花了任何来揭示所有从一开始就在他们的Qs中的约束？!？!？!!)但需要进行调整——单例集需要变成它们唯一元素的非集合(一个可怕的、可怕的、不好的规范，使得生成的字典几乎无法使用，并且让我强烈希望与那些似乎相信使令人作呕的糟糕规范改善了他们的教学，但是，这是另一个咆哮)。

无论如何，最好是添加一个后处理步骤:

def invert_dict(d):
    result = {}
    for k in d:
        if d[k] not in result:
            result[d[k]] = set()
        result[d[k]].add(k)
    return {k: d[k] if len(d[k])>1 else d[k].pop() for k in d}

没什么难的:只需“展开”单例就可以用 pop 设置为他们的一项。 (接下来是什么——另一个愚蠢的任意约束的迟来启示，例如“没有 if/else 表达式”？!-)

添加(保留上面的错误代码):需要在 return 语句中使用 result 而不是 d 清楚!即最后一行必须是

    return {k: result[k] if len(result[k])>1 else result[k].pop() for k in result}