蟒蛇：提取维数索引

Question

提取多维数组的一个片段我知道如何采取x[:,:,:,:,j,:]（它采用维度4的第j片）。

如果尺寸在运行时是已知的，并且不是已知的常量，是否有办法做同样的事情？

Answer 1

可以使用slice功能，并与调用它运行期间的适当变量列表如下：

# Store the variables that represent the slice in a list/tuple 
# Make a slice with the unzipped tuple using the slice() command 
# Use the slice on your array 

例子：

>>> from numpy import * 
>>> a = (1, 2, 3) 
>>> b = arange(27).reshape(3, 3, 3) 
>>> s = slice(*a) 
>>> b[s] 
array([[[ 9, 10, 11], 
     [12, 13, 14], 
     [15, 16, 17]]]) 

这是一样的：

>>> b[1:2:3] 
array([[[ 9, 10, 11], 
     [12, 13, 14], 
     [15, 16, 17]]]) 

最后，在平时的符号不指定2 :之间的任何东西相当于是把那些地方None元组中的你创建。

Answer 2

一个选择这样做，是构建编程切片：

slicing = (slice(None),) * 4 + (j,) + (slice(None),) 

另一种方法是使用numpy.take()或ndarray.take()：

>>> a = numpy.array([[1, 2], [3, 4]]) 
>>> a.take((1,), axis=0) 
array([[3, 4]]) 
>>> a.take((1,), axis=1) 
array([[2], 
     [4]]) 

Answer 3

您还可以使用省略号替换重复的冒号。 举例来说，请参阅answer至How do you use the ellipsis slicing syntax in Python?。