numpy.column_stack() in Python
numpy.column_stack() in Python
The numpy.column_stack() function stacks the 1-D arrays as columns into a 2-D array. It takes a sequence of 1-D arrays and stacks them as columns to make a single 2-D array.
Syntax
numpy.column_stack(tup)
Parameter
tup : This parameter represents a sequence of 1-D or 2-D arrays where all of them must have the same first dimension.
Return
This function returns the array formed by stacking the given arrays.
Example 1
# Python program explaining
# numpy.column_stack() function
import numpy as np
inp_arr1 = np.array(( 11, 12, 13 ))
print ("1st Input array: ", inp_arr1)
inp_arr2 = np.array(( 14, 15, 16 ))
print ("2nd Input array: ", inp_arr2)
# Stacking the two arrays
out_arr = np.column_stack((inp_arr1, inp_arr2))
print ("Output array: ", out_arr)
Output
1st Input array: [11 12 13] 2nd Input array: [14 15 16] Output array: [[11 14] [12 15] [13 16]]
Example 2
# Python program explaining
# numpy.column_stack() 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)
out_array = np.column_stack((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 12 13 14 15 16] [-11 -12 -13 -14 -15 -16]]
