a = np.array([[1, 2, 3], [4, 5, 6]]);
np.sum(a, axis = 0);

a = np.array([1, 2, 3]);
np.sum(a, axis = 0);