Practice>Python>Numpy>Sum and Prod

hackerrank.com

Sum and Prod

Problem link : https://www.hackerrank.com/challenges/np-sum-and-prod/problem

Difficulty : Easy

Tip

• Please understand how to use numpy.sum() and numpy.prod()
• They are all you should know

Solution

import numpy as np N, M = list(map(int, input().split(' '))) arrA = np.array([input().split(' ') for _ in range(N)], int) arrB = np.sum(arrA, axis=0) #print(arrB) print(np.prod(arrB, axis=0))