python - 复制嵌套列表并用连续数字填充

Question

给定这个嵌套列表:

foo = [["apple", "cherry"], ["banana"], ["pear", "raspberry", "pineapple"]]

我想保留结构并用连续的数字替换所有项目。我想要的输出是:

[[0, 1], [2], [3, 4, 5]]

我希望有一个简单的单行代码，但我想出的最短的工作解决方案是:

foo_numbers = []
count = 0

for i, sublist in enumerate(foo):
    foo_numbers.append([])
    for item in sublist:
        foo_numbers[i].append(count)
        count += 1

通常这些手动迭代器表明有更 pythonic 的方法来实现同样的事情。如果这是通过列表理解来完成的，我看不出如何为两个循环组成一个“共享计数器”，所以它不会对每个 sublist 从零开始。

Answer 1

您可以使用 itertools.count使用嵌套列表理解:

from itertools import count

foo = [["apple", "cherry"], ["banana"], ["pear", "raspberry", "pineapple"]]

c = count()  # 0 start is default, e.g. count(1) will start from 1
res = [[next(c) for _ in lst] for lst in foo]

print(res)
# [[0, 1], [2], [3, 4, 5]]