Numpy!

snehalmastud · Feb 23, 2020 · c7c21ad · c7c21ad
1 parent 9b282a1
commit c7c21ad
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Numpy/Dot and Cross/Solution.py b/Numpy/Dot and Cross/Solution.py
@@ -0,0 +1,10 @@
+import numpy
+
+N = int(input())
+
+A = numpy.array([list(map(int, input().split())) for n in range(N)])
+B = numpy.transpose(numpy.array([list(map(int, input().split())) for n in range(N)]))
+
+AB = numpy.array([[numpy.dot(A[i], B[j]) for j in range(N)] for i in range(N)])
+
+print (AB)