w3resource
NumPy Exercises

NumPy: Compute sum of all elements, sum of each column and sum of each row of an given array

NumPy: Basic Exercise-32 with Solution

Write a NumPy program to compute sum of all elements, sum of each column and sum of each row of an given array.

Sample Solution :

Python Code :

import numpy as np
x = np.array([[0,1],[2,3]])
print("Original array:")
print(x)
print("Sum of all elements:")
print(np.sum(x))
print("Sum of each column:")
print(np.sum(x, axis=0))
print("Sum of each row:")
print(np.sum(x, axis=1))

Sample Output:

Original array:
[[0 1]
 [2 3]]
Sum of all elements:
6
Sum of each column:
[2 4]
Sum of each row:
[1 5]                         

Pictorial Presentation:

NumPy: Compute sum of all elements, sum of each column and sum of each row of an given array.

Python Code Editor:


Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a NumPy program to create a 3x3x3 array filled with arbitrary values.
Next: Write a NumPy program to compute the inner product of two given vectors.

What is the difficulty level of this exercise?