Numpy!

snehalmastud · Feb 23, 2020 · 8aced0a · 8aced0a
1 parent 5ccda2b
commit 8aced0a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Numpy/Sum and Prod/Solution.py b/Numpy/Sum and Prod/Solution.py
@@ -0,0 +1,10 @@
+import numpy
+
+lines = []
+n, m = map(int, raw_input().split())
+for _ in range(n):
+    lines.append(map(int, raw_input().split()))
+
+my_array = numpy.array(lines)
+
+print numpy.prod(numpy.sum(my_array, axis = 0))