pythonのlistの取り扱い方

Question

適切なタイトルが思いつかずすみません。

idx_list = [0, 1, 0, 2]
matrix = [
    [0.1, 0.4, 0.3, 0.5],
    [0., 0., 0., 0.],
    [0.1, 0.1, 0.2, 0.3],
]

二つの配列があるときに

[
    [0.1, 0.4, 0.3, 0.5],
    [0., 0., 0., 0.],
    [0.1, 0.4, 0.3, 0.5],
    [0.1, 0.1, 0.2, 0.3]
]

のように、idx_listの添字に応じたmatrixの要素を取りたいのですが、numpyなどを使って効率的に行うことは可能でしょうか？
本来はidx_listは多次元配列での挙動を想定しています。

以下のようにすることは思いついたのですが、多次元の場合for文が続くためあまり望ましくないと思っています。

res = []
for i in range(len(idx_list)):
    res.append(matrix[idx_list[i]])
print(res)

よろしくお願いします。

Answer 1

すみません、普通にできました。
https://note.nkmk.me/python-numpy-fancy-indexing/