python - 在不循环的情况下访问 3D numpy 数组中的多列

Question

我有一个大型 3D 阵列 A带形状(N, M, L) .
我有一个列的坐标列表，我想访问存储在二维数组中 B :

[[i1 j1]
 [i2 j2]
 [i3 j3]
 ....   ]

我有一些工作正常但涉及循环的东西 B并访问 A多次。有没有办法使用切片或其他方法来避免这种情况？
到目前为止我的代码:

data_out = []
for p in B:
    i, j = p
    col = A[:, i, j]
    data_out.append(col)

Answer 1

使用花哨的索引:

A[(slice(None), *B.T)].T

使用星型扩展需要显式括号，这意味着您必须写出 :明确为 slice(None) .你也可以这样做

A[:, B[:, 0], B[:, 1]].T