numpy.stack

Cette fonction joint la séquence de tableaux le long d'un nouvel axe. Cette fonction a été ajoutée depuis la version 1.10.0 de NumPy. Les paramètres suivants doivent être fournis.

Note- Cette fonction est disponible à partir de la version 1.10.0 .

numpy.stack(arrays, axis)

Où,

N ° Sr. Paramètre et description
1

arrays

Séquence de tableaux de même forme

2

axis

Axe dans le tableau résultant le long duquel les tableaux d'entrée sont empilés

Exemple

import numpy as np 
a = np.array([[1,2],[3,4]]) 

print 'First Array:' 
print a 
print '\n'
b = np.array([[5,6],[7,8]]) 

print 'Second Array:' 
print b 
print '\n'  

print 'Stack the two arrays along axis 0:' 
print np.stack((a,b),0) 
print '\n'  

print 'Stack the two arrays along axis 1:' 
print np.stack((a,b),1)

Il devrait produire la sortie suivante -

First array:
[[1 2]
 [3 4]]

Second array:
[[5 6]
 [7 8]]

Stack the two arrays along axis 0:
[[[1 2]
 [3 4]]
 [[5 6]
 [7 8]]]

Stack the two arrays along axis 1:
[[[1 2]
 [5 6]]
 [[3 4]
 [7 8]]]