使用容器（例如元组或列表）进行Numpy切片

Question

我正在尝试将一个nDnumpy.ndarray切片。为了简单起见，假设它是3D（但实际上它可能是4D，5D，等等），并且由nx，ny，nz维度组成

我想使用容器（例如元组或列表）来选择ny和nz，提取给定点ny，nz的所有nx行

例如：

假设我有一个维度为：（46841128,20）的3DarrayX，我有一个container = (127,11)

理想的解决方案：

x_selected = X[:,container]

其中x_selected应该有维度（46841，）

Answer 1

您只需使用以下功能：

x_selected=X[:,container[0],container[1]]

我试过这个：

A=np.zeros((3,5,7))
A[1,2,:].shape

我得到（7，）作为输出

Answer 2

x_selected = X[(slice(None),) + container]

因此，您可以使用包含slice和整数混合的任何元组对X数组进行索引slice只是一个纯python对象described here。此用于索引的元组的元素不应超过X数组的维度