反向键(由列表组成)和Python字典中的值

时间:2012-06-01 22:03:14

标签: python dictionary list-comprehension state-machine

我一直试图从这里的其他帖子中解决这个问题,但不能。

我有一个Python词典

old_dict = { (1,'a') : [2],
          (2,'b') : [3,4],
          (3,'x') : [5],
          (4,'y') : [5],
          (5,'b') : [3,4], 
          (5,'c') : [6],
          }

我需要反转这一点,以便结果我会:

new_dict = { (6,'c') : [5],
          (5,'x') : [3],
          (5,'y') : [4],
          (4,'b') : [5, 2],
          (3,'b') : [5, 2], 
          (2,'a') : [1],
          }

(这描述了有限状态机的边缘,我需要向后运行它:它必须像之前那样接受反向输入)

例如,在old_dict中,第一个键是列表(1, 'a') : [2],现在,这个键应该变为(2, 'a'), [1] ...或(4,'y') : [5]变为(5,'y') : [4]等。 - 我希望我的意思是可以理解的。

我一直试图用列表推导来解决这个问题,但还没有成功。

更新:我尝试了F.C.的建议,但不知怎的,我无法让代码工作。我把它插入一个函数中,如下所示:

old_dict1 = { (1,'a') : [2],
          (2,'b') : [3,4],
          (3,'x') : [5],
          (4,'y') : [5],
          (5,'b') : [3,4], 
          (5,'c') : [6],
          }

def reverse_dict(old_dict):
    new_dict = {}
    add_to_dict = new_dict.setdefault

    map(lambda kv: add_to_dict(kv[0], []).append(kv[1]),   
        sum([[((x, k[1]), k[0]) for x in v] for k, v in old_dict.items()],
            []))        # sum will take this to start adding
    return new_dict

new_dict1 = reverse_dict(old_dict1)

print(new_dict1)

但我只返回一个空字典{}

我做错了吗? (我对Python知之甚少,所以如果我犯了一个太傻的错误,请原谅我......)

3 个答案:

答案 0 :(得分:8)

这很复杂,我不打扰列表推导。另外,我假设你不是在寻找任何严格的价值表。

new_dict = {}
for k, vals in old_dict.items():
    k_num, k_char = k
    for num in vals:
        new_dict.setdefault((num, k_char), []).append(k_num)

或使用defaultdict

new_dict = collections.defaultdict(list)
for k, vals in old_dict.items():
    k_num, k_char = k
    for num in vals:
        new_dict[(num, k_char)].append(k_num)

对于那些有兴趣尽可能简洁的人来说,我发现这个更加压缩的版本也是一个选项。从可读性的角度来看,我不确定我对此的看法,所以我更改了变量名称以便更清晰:

new_dict = collections.defaultdict(list)
for (num_in, char_in), nums_out in old_dict.items():
    for num_out in nums_out:
        new_dict[(num_out, char_in)].append(num_in)

答案 1 :(得分:0)

这适用于您的数据,将其视为丑陋的黑客只是为了好玩

最好在更多的代码行中这样做,这样可以更容易理解,但有时我无法抗拒写这些装置的诱惑。

希望它有所帮助。

def reverse_dict(old_dict):
    """
    >>> sorted(reverse_dict({(1,'a'): [2],
    ...               (2,'b'): [3,4],
    ...               (3,'x'): [5],
    ...               (4,'y'): [5],
    ...               (5,'b'): [3,4], 
    ...               (5,'c'): [6],
    ...              }).items())
    [((2, 'a'), [1]), ((3, 'b'), [2, 5]), ((4, 'b'), [2, 5]), ((5, 'x'), [3]), ((5, 'y'), [4]), ((6, 'c'), [5])]
    """
    new_dict = {}
    add_to_dict = new_dict.setdefault       # you could use a [defaultdict][1] instead

    map(lambda kv: add_to_dict(kv[0], []).append(kv[1]),   # if kv[0] not in dict get [] and add to it
        sum([[((x, k[1]), k[0]) for x in v] for k, v in old_dict.items()],
            []))        # sum will take this to start adding
    return new_dict

要测试代码,只需将其复制到文件so.py并按以下方式运行:

$ python -m doctest so.py -v
Trying:
    sorted(reverse_dict({(1,'a'): [2],
                  (2,'b'): [3,4],
                  (3,'x'): [5],
                  (4,'y'): [5],
                  (5,'b'): [3,4], 
                  (5,'c'): [6],
                 }).items())
Expecting:
    [((2, 'a'), [1]), ((3, 'b'), [2, 5]), ((4, 'b'), [2, 5]), ((5, 'x'), [3]), ((5, 'y'), [4]), ((6, 'c'), [5])]
ok
1 items had no tests:
    so
1 items passed all tests:
   1 tests in so.reverse_dict
1 tests in 2 items.
1 passed and 0 failed.
Test passed.

它使用doctest来更轻松地测试它是否符合您的要求。

答案 2 :(得分:-1)

我认为您对dict目的的理解存在问题。 dict数据结构不应被视为以任何特定方式排序,因为它使用哈希表来访问元素。您应该阅读docs.items() here中有关实施细节的说明也很重要。事实证明,实现可能会给你你似乎期待的订单,但你不应该指望它。

如果订单对您很重要,那么您应该至少在代码中重要的部分使用list。使用.items()上的dict方法获取(key,value)对的列表,然后您可以使用列表中的常用排序方法以任意方式对它们进行排序。