numpy column code example

Example 1: print column in 2d numpy array

import numpy as np
x=np.arange(25).reshape(5,5)
print(x)
#print(x[:,index_of_column_you_need])
print(x[:,3])

Example 2: get column or row of matrix array numpy python

test = numpy.array([[1, 2], [3, 4], [5, 6]])
column = test[:,0]
# column == array([1, 3, 5])
row = test[1,:]
# row == array([3, 4])

Example 3: numpy column

>>> test[:,0]
array([1, 3, 5])