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]。
