Python: list of list, 将内部 list 的 index 作为该内部 list 中每个元素的分类标签

一、需求

输入 a = [[0], [3,4], [1,2]]，要求输出为 [0, 2, 2, 1, 1]

解释：[0] 是第 0 位，所以 0 位置的输出为0，[1,2] 的 index= 2，所以 1 和 2  对应位置的输出是2

二、代码

a = [[0],[3,4],[1,2]]

max_len = 5

rs = [0] * max_len

for i,x in enumerate(a):
# x 分别为 [0],[3,4],[1,2]

    for idx in x: 
# idx 为 list x 中的每个元素

        rs[idx] = i

此时 rs = [0, 2, 2, 1, 1]。