Problem
dot
The dot tool returns the dot product of two arrays.
import numpyA = numpy.array([ 1, 2 ])B = numpy.array([ 3, 4 ])print numpy.dot(A, B) #Output : 11
cross
The cross tool returns the cross product of two arrays.
import numpyA = numpy.array([ 1, 2 ])B = numpy.array([ 3, 4 ])print numpy.cross(A, B) #Output : -2
Task
You are given two arrays A and B. Both have dimensions of NXN.
Your task is to compute their matrix product.
Input Format
The first line contains the integer N.
The next N lines contains N space separated integers of array A.
The following N lines contains N space separated integers of array B.
Output Format
Print the matrix multiplication of A and B.
Sample Input
21 23 41 23 4
Sample Output
[[ 7 10] [15 22]]
Solution – Dot and Cross In Python | HackerRank
import numpy as npn = int(input())a = np.array([list(map(int, input().split())) for _ in range(n)])b = np.array([list(map(int, input().split())) for _ in range(n)])print(np.dot(a, b))
Comments
Post a Comment