Dot and Cross in numpy Hackerrank
The dot tool returns the dot product of two arrays.
import numpy
A = numpy.array([ 1, 2 ])
B = numpy.array([ 3, 4 ])
print numpy.dot(A, B) #Output : 11
The cross tool returns the cross product of two arrays.
import numpy
A = numpy.array([ 1, 2 ])
B = numpy.array([ 3, 4 ])
print numpy.cross(A, B) #Output : -2
Task
You are given two arrays and . Both have dimensions of X.
Your task is to compute their matrix product.
Input Format
The first line contains the integer .
The next lines contains space separated integers of array .
The following lines contains space separated integers of array .
Output Format
Print the matrix multiplication of and .
Sample Input
2
1 2
3 4
1 2
3 4
Sample Output
[[ 7 10] [15 22]]
Solution :
import numpy
m =int(input())
arr1 = numpy.array([input().split() for i in range(m)],int)
arr2 = numpy.array([input().split() for i in range(m)],int)
print(numpy.dot(arr1,arr2))
Comments
Post a Comment