如何在python Classlis中高效地生成编号列表

Question

如何使下面的代码更加紧凑和高效。你知道吗

在这里，代码用于查找列表中某个数值所在的位置。 例如，给定的一组数字

ListNo = [[100,2,5], [50,10], 4, 1, [6,6,500]]

100、50和500分别位于0、3和9的位置。你知道吗

测试代码如下

ListNo = [[100,2,5], [50,10], 4, 1, [6,6,500]]
NumberedList = ListNo
Const = 0
items = 0

for i, item in enumerate(ListNo):
    MaxRange = len(item) if isinstance(item, list) else 1

    for x in range(0, MaxRange):
        if MaxRange > 1:
            NumberedList[i][x] = Const
        else:
            NumberedList[i] = Const
        Const = Const + 1

print(NumberedList)

[[0, 1, 2], [3, 4], 5, 6, [7, 8, 9]]

我的问题是，是否有另一种方法可以使这段代码更加紧凑和高效。你知道吗

Answer 1

您可以使用itertools.count：

from itertools import count
i = count()
print([[next(i) for _ in range(len(l))] if isinstance(l, list) else next(i) for l in ListNo])

这将输出：

[[0, 1, 2], [3, 4], 5, 6, [7, 8, 9]]