我一直试图从这里的其他帖子中解决这个问题,但不能。
我有一个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知之甚少,所以如果我犯了一个太傻的错误,请原谅我......)
答案 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)