Ritagliare parte di np.array

Ihave un array numpy come

A.shape
(512,270,1,20)

Non voglio utilizzare tutti i 20 livelli in dimensione 4. Il nuovo array dovrebbe essere come

Anew.shape
(512,270,1,2)

Quindi voglio ritagliare 2 “fette” della matrice di Un

InformationsquelleAutor refle | 2015-10-20

 

2 Replies
  1. 8

    Da la documentazione di python, la risposta è:

    start = 4 # Index where you want to start.
    Anew = A[:,:,:,start:start+2]
  2. 3

    È possibile utilizzare un elenco o un array di indici, piuttosto che la fetta di notazione per selezionare una sequenza arbitraria di indici in finale dimensione:

    x = np.zeros((512, 270, 1, 20))
    y = x[..., [4, 10]] # the 5th and 11th indices in the final dimension
    print(y.shape)
    # (512,270,1,2)

Lascia un commento