Сопоставить значения в списке со списками по признакам

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)