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

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数组（ndarray）。

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)