Sum And Prod Hackerrank Solution Numpy Python