重新排列字典，使值成为下一个键

Question

我有一本字典如下：

{1: 3, 2: 4, 3: 5, 4: 1, 5: 2}

我想重新排列它，使当前对的值是下对的键，像这样：

{1: 3, 3: 5, 5: 2, 2: 4, 4: 1}

我该如何处理？

Answer 1

您可以使用 itertools 中的累积来做到这一点：

d = {1: 3, 2: 4, 3: 5, 4: 1, 5: 2}

from itertools import accumulate
od = {k:d[k] for k in accumulate(d,lambda k,_:d[k]) }

#od : {1: 3, 3: 5, 5: 2, 2: 4, 4: 1}

Answer 2

这可以使用以下代码 - 从 Python 3.7（CPython 的 Python 3.6）开始的字典按插入排序，因此我们从 d 的第一个键开始，并迭代，更新下一个键以检查使用最后一个键的值，直到新字典的长度与旧字典的长度相同。

代码：

d = {1: 3, 2: 4, 3: 5, 4: 1, 5: 2}

curr = list(d.keys())[0]

d_new = {}

while len(d) > len(d_new):
    d_new[curr] = curr = d[curr]

输出：

>>> d_new
{1: 3, 3: 5, 5: 2, 2: 4, 4: 1}

在 Python 3.7 以下，您可以使用OrderedDict：

from collections import OrderedDict

d = {1: 3, 2: 4, 3: 5, 4: 1, 5: 2}

curr = 1

d_new = OrderedDict()

while len(d) > len(d_new):
    d_new[curr] = curr = d[curr]

根据下面的 cmets，如果键/值包含循环，我们希望移动到下一个键值对。这可以通过以下方式实现：

代码：

d = {1: 3, 2: 4, 3: 1, 4: 1, 5: 2}

curr = list(d.keys())[0]

d_new = {}

while len(d) > len(d_new):
    if curr not in d_new:
        d_new[curr] = curr = d[curr]
    else:
        curr = next(x for x in d.keys() if x not in d_new)

输出：

>>> d_new
{1: 3, 3: 1, 2: 4, 4: 1, 5: 2}