-
Notifications
You must be signed in to change notification settings - Fork 64
/
Copy pathnumpy_sum_prod.py
63 lines (41 loc) · 1.55 KB
/
numpy_sum_prod.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
"""
Problem Statement
sum
The sum tool returns the sum of array elements over a given axis.
import numpy
my_array = numpy.array([ [1, 2], [3, 4] ])
print numpy.sum(my_array, axis = 0) #Output : [4 6]
print numpy.sum(my_array, axis = 1) #Output : [3 7]
print numpy.sum(my_array, axis = None) #Output : 10
print numpy.sum(my_array) #Output : 10
By default, the axis value is None. Therefore, it performs a sum over all the dimensions of the input array.
prod
The prod tool returns the product of array elements over a given axis.
import numpy
my_array = numpy.array([ [1, 2], [3, 4] ])
print numpy.prod(my_array, axis = 0) #Output : [3 8]
print numpy.prod(my_array, axis = 1) #Output : [ 2 12]
print numpy.prod(my_array, axis = None) #Output : 24
print numpy.prod(my_array) #Output : 24
By default, the axis value is None. Therefore, it performs the product over all the dimensions of the input array.
Task
You are given a 2-D array with dimensions NXM.
Your task is to perform the sum tool over axis 0 and then find the product of that result.
Input Format
The first line of input contains space separated values of N and M.
The next N lines contains M space separated integers.
Output Format
Compute the sum along axis 0. Then, print the product of that sum.
Sample Input
2 2
1 2
3 4
Sample Output
24
"""
import numpy
k = map(int,raw_input().split())
n = [ map(int,raw_input().split()) for _ in xrange(k[0])]
a = numpy.array(n)
s = numpy.sum(a,axis=0)
print numpy.prod(s,axis=0)