如何在遍历字典的同时，迭代剩余的项？

Question

我有一个字典，想要在遍历它的时候更优雅地迭代其后续的键，如下所示：

d = {"a": 1, "b": 2, "c": 3}

keys = list(d.keys())

for i in range(len(keys)):
    print(keys[i] + ":")
    for j in range(i+1, len(keys)):
        print("	" + keys[j])

这段代码会输出：

a:
    b
    c
b:
    c
c:

感觉可以利用迭代器来实现这一目标，有没有更简洁的方法呢？

Konrad Rudolph · Answer

d = { “a”: 1, “b”: 2, “c”: 3 }
k = list(d)
for i in range(len(k)):
    print(k[i], k[i+1:])

Duncan · Answer

由于顺序无关紧要，您可以每次都弹出一个键，并获取剩余的键：

d = {"a": 1, "b": 2, "c": 3}
temp_d = d.copy()  # 创建原字典的副本
while temp_d:
    key = temp_d.popitem()[0]  # 弹出一个键值对，并只取键部分
    print(key + ":", temp_d.keys())  # 输出已弹出键及剩余键集合

如果你不再需要原始字典，就可以直接操作 d 而无需创建副本。

Donal Fellows · Answer

有两种基于迭代器的解决方案...

第一个方案明确创建了一个迭代器，以便我们能够访问它（不同于使用 for k in d: 时隐式使用的迭代器）。通过 for k in keys: 循环获取每个键之后，该迭代器就变成了剩余键的迭代器。通过浅复制，我们可以在不干扰主迭代器且不复制字典内容的情况下遍历这些剩余键。

from copy import copy

d = {"a": 1, "b": 2, "c": 3}

keys = iter(d)
for k in keys:
    print(k + ":")
    for k in copy(keys):
        print("	" + k)

输出（在线尝试）：

a:
    b
    c
b:
    c
c:

另一种替代方案，使用 islice（在线尝试）：

from itertools import islice

d = {"a": 1, "b": 2, "c": 3}

for i, k in enumerate(d, 1):
    print(k + ":")
    for k in islice(d, i, None):
        print("	" + k)

每一次 islice 都通过快速遍历前 i 个键来“跳过”它们，因此总体上有些浪费双层迭代。但是相比 print 操作，这种方式相当快速。

还有另外两种解决方案，虽然它们确实复制了字典内容，但比你提供的代码更简短。

d = {"a": 1, "b": 2, "c": 3}

ks = list(d)
for i, k in enumerate(ks, 1):
    print(k + ":")
    for k in ks[i:]:
        print("	" + k)

d = {"a": 1, "b": 2, "c": 3}

ks = list(d)
while ks:
    print(ks.pop(0) + ":")
    for k in ks:
        print("	" + k)

后一种方法受到了 Guy的回答启发。

Shog9 · Answer

对于基于迭代器的方法，你可以使用 itertools 中的 tee 函数来“快照”字典键的迭代器。

from itertools import tee

d = {"a": 1, "b": 2, "c": 3}

iterd = iter(d)
for k in d:
    print(k + ":")
    next(iterd)
    iterd, rest = tee(iterd)
    for k in rest:
        print("	", k)

# 输出：
# a:
#      b
#      c
# b:
#      c
# c:

# 注释：尽管这个解决方案在概念上很有趣，但从效率角度来看可能相当低效

一个更高效的解决方案是使用队列：

from collections import deque

d = {"a": 1, "b": 2, "c": 3}

qd = deque(d)
while qd:
    print(qd.popleft() + ":")  
    for k in qd:
        print("	", k)

或者简单地使用列表（无需删除或弹出元素）：

d = {"a": 1, "b": 2, "c": 3}

ld = list(d)
for i, k in enumerate(ld, 1):
    print(k + ":")  
    for j in range(i, len(d)):
        print("	", ld[j])