Indexation dimensionnelle

Numpy prend en charge un outil extrêmement puissant - l'indexation des dimensions.

Pour indexer une dimension, nous utilisons les deux-points :. Les deux-points récupèrent tous les éléments à l'index spécifique où ils sont placés. Par exemple, supposons que nous ayons un tableau avec la forme ary.shape # --> (2, 1, 3, 1), placer les deux-points : à ary[:, 0, 0, 0] prendra les deux éléments les plus à gauche de la forme et pour chacun d'eux, il extraira le zéroième élément pour le reste des dimensions.

Note : L'emplacement des deux-points dans les crochets [] correspond à la dimension que nous indexons, ce qui signifie :

• ary[:, 0, 0, 0] Indexe la dimension 4D
• ary[0, :, 0, 0] Indexe la dimension 3D
• ary[0, 0, :, 0] Indexe la dimension 2D
• ary[0, 0, 0, :] Indexe la dimension 1D
• ary[:, :, 0, 0] Indexe les dimensions 4D et 3D
• etc

Exemple 1

ary = np.array([
	[1, 2], 
	[3, 4]
])
ary[:, 0] # --> [1, 3]
ary[0, :] # --> [1, 2]

Les deux-points prennent tous les éléments à l'index spécifié et continuent vers l'index suivant

Dans notre cas : 

• ary[:, 0] Nous prenons tous les premiers éléments et à chaque élément nous prenons l'élément 0
• ary[0, :] Nous prenons le premier élément et dans cet élément nous prenons tous les éléments.

Exemple 2

Tableau 3D

ary = np.array([
	[[1, 2], [3, 4]],
	[[5, 6], [7, 8]]
])
ary[:, 0, :] # --> [[1, 2], [5, 6]]
ary[:, :, 0] # --> [[1, 3], [5, 7]]

• ary[:, 0, :] Nous prenons tous les éléments, puis nous prenons le premier élément et enfin nous prenons tout ce qui s'y trouve
• ary[:, :, 0] Nous prenons tous les éléments, puis nous prenons tous ces éléments et enfin nous prenons le premier élément de chacun.
• Pouvez-vous deviner quel est le résultat pour ary[0, :, :]

Exemple 3

Obtenir le premier et le dernier index de chaque ligne

x = np.array([
	[ 0,  1,  2,  3,  4,  5],
	[ 6,  7,  8,  9, 10, 11],
	[12, 13, 14, 15, 16, 17],
	[18, 19, 20, 21, 22, 23],
])
x[:, np.array([0, -1])]
>>> [0, 5, 6, 11, 12, 17, 18, 23]

Exemple 4

Un tableau avec la forme (2, 1, 3, 1) tout comme dans l'exemple d'ouverture

ary = np.array([
    [
        [
            [1],[2],[3]
        ]
    ],
    [
        [
            [4],[5],[6]
        ]
    ]
])
ary[:, 0, 0, 0] # --> [1, 4]
ary[0, :, 0, 0] # --> [1]
ary[0, 0, :, 0] # --> [1, 2, 3]
ary[0, 0, 0, :] # --> [1]

Note : il est également possible d'utiliser la syntaxe de découpage (slicing) : start:stop:step que nous avons apprise lors de la leçon sur le Slicing

• ary[:, 0:5, 2:3, :]