python: estrazione di una porzione di una matrice multidimensionale data l'indice di dimensione

Question

So come utilizzare lo x[:,:,:,:,j,:] (che accetta la sezione jth della dimensione 4).

C'è un modo per fare la stessa cosa se la dimensione è nota al runtime e non è una costante conosciuta?

Answer 1

È possibile utilizzare la funzione slice e chiamarlo con il elenco delle variabili appropriato durante il runtime nel modo seguente:

# 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 

Esempio:

>>> 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]]]) 

Questa è la stessa:

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

Infine, l'equivalente di non specificare nulla tra 2 : nella consueta notazione è di mettere None in quei luoghi nella tupla si creare.

Answer 2

Una possibilità per farlo è quello di costruire l'affettamento programatically:

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

Un'alternativa è quella di utilizzare numpy.take() o 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

È inoltre possibile utilizzare i puntini di sospensione per sostituire i due punti ripetuti. Vedere un answer a How do you use the ellipsis slicing syntax in Python? per un esempio.