python - 按索引将列表中的值映射到列表

Question

我想将列表中的点映射到基于另一个列表中的索引的列表。例如，假设我们有:

k: 2
a: [0,1,2,1]
b: [[i,...], [t,...], [u,...], [v,...]]

我想将 b 中的值映射到从 a 到 c 的索引。

k 是 a 中的最大值。我仅使用 i,t,u,v 来标识 b 的子数组。

c: [
    [[i,...]],
    [[t,...], [v,...]],
    [[u,...]]
   ]

任何帮助将不胜感激 a、b 和 c 是 numpy 数组 (ndarrays)。

Answer 1

def test():
    a = [0,1,2,1]
    k = max(a)
    b = [[1,2,3], [4,5], [6,7], [8, 9]]
    c = [[] for i in range(k+1)]
    for i in range(len(b)):
        c[a[i]].append(b[i])
    print(c)