python - 独立到列的 Numpy 索引

Question

假设我有一个像

arr = np.arange(27).reshape(9,3)

我有一个这样的索引列表：

idx = np.array([[0, 0, 0],
                [1, 0, 0],
                [1, 7, 1],
                [5, 6, 7]])

哪里idx.shape[1] == arr.shape[1]。我想要返回的是：

out = np.array([[ 0,  1,  2],
                [ 3,  1,  2],
                [ 3, 22,  5],
                [15, 19, 23]])

也就是说，一个数组，其中一些(i, j), out[i,j] == arr[idx[i,j], j]。