Concaténer et diviser des tableaux
NumPy fournit deux fonctions pour concaténer des tableaux :
np.concatenate
: pour concaténer des tableaux le long d'un axe donné
np.stack
: pour concaténer des tableaux le long d'un nouvel axe
Vous pouvez diviser des tableaux à l'aide de la fonction np.split
.
Concaténer des tableaux
Créez deux tableaux a
et b
à titre d'exemple :
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
Utiliser concatenate
Vous pouvez concaténer ces tableaux le long du premier axe (0) à l'aide de la fonction np.concatenate
:
c = np.concatenate((a, b))
print(c)
Sortie :
[1 2 3 4 5 6]
Utiliser stack
Vous pouvez également concaténer ces tableaux le long d'un nouvel axe à l'aide de la fonction np.stack
:
d = np.stack((a, b))
print(d)
print(d.shape)
Sortie :
[[1 2 3]
[4 5 6]]
(2, 3)
Diviser des tableaux
Créez un tableau a
de forme (6,) à titre d'exemple :
a = np.array([1, 2, 3, 4, 5, 6])
Utiliser split
Vous pouvez diviser ce tableau en deux tableaux de longueur 3 à l'aide de la fonction np.split
:
b, c = np.split(a, 2)
print(b)
print(c)
Sortie :
[1 2 3]
[4 5 6]