numpy.squeeze

Cette fonction supprime l'entrée unidimensionnelle de la forme du tableau donné. Deux paramètres sont nécessaires pour cette fonction.

numpy.squeeze(arr, axis)

Où,

N ° Sr.	Paramètre et description
1	arr Tableau d'entrée
2	axis int ou tuple de int. sélectionne un sous-ensemble d'entrées unidimensionnelles dans la forme

Exemple

import numpy as np  
x = np.arange(9).reshape(1,3,3) 

print 'Array X:' 
print x 
print '\n'  
y = np.squeeze(x) 

print 'Array Y:' 
print y 
print '\n'  

print 'The shapes of X and Y array:' 
print x.shape, y.shape

Sa sortie est la suivante -

Array X:
[[[0 1 2]
 [3 4 5]
 [6 7 8]]]

Array Y:
[[0 1 2]
 [3 4 5]
 [6 7 8]]

The shapes of X and Y array:
(1, 3, 3) (3, 3)