xxxxxxxxxx
axis = 1 => 'Coulumn'
axis = 0 => 'Row'
+------------+---------+--------+
| | A | B |
+------------+---------+---------
| 0 | 0.626386| 1.52325|----axis=1----->
+------------+---------+--------+
. | |
. | axis=0 |
↓ ↓
xxxxxxxxxx
import numpy as np
# Sample 2D array
arr = np.array([[1, 2, 3],
[4, 5, 6]])
column_sum = np.sum(arr, axis=0)
print(column_sum)