sum along axis python code example
Example 1: sum axis in python
import numpy as np
array1 = np.array(
[[1, 2],
[3, 4],
[5, 6]])
total_0_axis = np.sum(array1, axis=0)
print(f'Sum of elements at 0-axis is {total_0_axis}')
total_1_axis = np.sum(array1, axis=1)
print(f'Sum of elements at 1-axis is {total_1_axis}')
Output:
Sum of elements at 0-axis is [ 9 12]
Sum of elements at 1-axis is [ 3 7 11]
Example 2: sum along axis python
import numpy as np
matrix=np.ones((10,10))
print(matrix.sum(axis=0))
print(matrix.sum(axis=1))