Python Numpy中的高效矩阵索引

我有一个大小为256 X 16的矩阵

a

，还有一个大小为8 X 8的矩阵

B

，其中

B

的每个元素都是

[0,15]

范围内的整数。如果A只是一个行向量（即大小为1X16），则

C=A[B]

将输出一个8X8矩阵，其中

Cij=A[bij]

。如果

A

是256X16，使用numpy（或其他任何东西），有没有什么有效的方法可以做到同样的事情？我的意思是我想要一个大小为256 X 8 X 8的输出矩阵

C

（即

B

正在索引

A

的每一行）。

---

谢谢

好吧，我想我已经解决了：

C = A[:, B]