numpy.dstack() in Python

numpy.dstack() in Python

The numpy.dstack() function stacks the arrays in sequence depth wise (along third axis).

Syntax

numpy.dstack(tup)

Parameter

tup: This parameter represents the sequence of arrays where the arrays must have the same shape along all except the third axis.

Return

This function returns the array formed by stacking the given arrays, will be at least 3-D.

Example 1

# Python program explaining
# numpy.dstack() function
import numpy as np
inp_array1 = np.array([[ 11, 12, 13], [ -11, -12, -13]] )
print ("Input array: ", inp_array1) 
inp_array2 = np.array([[ 14, 15, 16], [ -14, -15, -16]] )
print ("Input array: ", inp_array2) 
#stack arrays depth wise (along third axis)
out_array = np.dstack((inp_array1, inp_array2))
print ("Output array: ", out_array)

Output

Input array:  [[ 11  12  13]
[-11 -12 -13]]
Input array:  [[ 14  15  16]
[-14 -15 -16]]
Output array:  [[[ 11  14]
[ 12  15]
[ 13  16]]
[[-11 -14]
[-12 -15]
[-13 -16]]]