numpy column sum code example

Example: how to to get sum of column or row in numpy

np.sum([[0, 1], [0, 5]], axis=0)
#array([0, 6])

np.sum([[0, 1], [0, 5]], axis=1)
#array([1, 5])