numpy.stack() in Python
The numpy.stack() function joins a sequence of arrays along a new axis.
Syntax
numpy.stack(arrays, axis=0, out=None)
Parameter
arrays: This parameter represents a sequence of the array where each array have the same shape.
axis: It is an optional parameter which takes integer values. It represents the axis along which the arrays will be joined.
out: This parameter represents the destination to place the result.
Return
This function returns the stacked array which has one or more dimension than the input arrays.
Example 1
# Python program explaining
# numpy.stack() function
import numpy as np
# input array
inp_arr1 = np.array([ 5, 6, 7] )
print ("1st Input array: ", inp_arr1)
inp_arr2 = np.array([ 10, 11, 12] )
print ("2nd Input array: ", inp_arr2)
# Stacking the arrays along axis 0
out_arr = np.stack((inp_arr1, inp_arr2), axis = 0)
print ("Output array along axis 0:\n ", out_arr)
Output
1st Input array:
[5 6 7]
2nd Input array:
[10 11 12]
Output array along axis 0:
[[5 6 7]
[10 11 12]]