Answers for "sum accros axis numpy"

3

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]
Posted by: Guest on June-24-2020
1

sum along axis python

import numpy as np
matrix=np.ones((10,10))
print(matrix.sum(axis=0))
print(matrix.sum(axis=1))
Posted by: Guest on March-21-2020

Python Answers by Framework

Browse Popular Code Answers by Language