HackerRank Array Mathematics problem solution in python

In this Array Mathematics problem You are given two integer arrays, A and B of dimensions N X M. Your task is to perform the following operations:

• Add (A + B)
• Subtract (A - B)
• Multiply (A * B)
• Integer Division (A / B)
• Mod (A % B)
• Power (A ** B)

Problem solution in Python 2 programming.

import numpy
l=map(int,raw_input().split())
a=[]
b=[]
for i in range(l[0]):
    a.append(map(int,raw_input().split()))
for i in range(l[0]):
    b.append(map(int,raw_input().split()))
a=numpy.array(a)
b=numpy.array(b)
print numpy.add(a, b)
print numpy.subtract(a, b)
print numpy.multiply(a, b)
print numpy.divide(a, b)
print numpy.mod(a, b)
print a**b 

Problem solution in Python 3 programming.

import numpy as np
n, m = map(int, input().split())
a, b = (np.array([input().split() for _ in range(n)], dtype=int) for _ in range(2))
print(a+b, a-b, a*b, a//b, a%b, a**b, sep='\n')

Problem solution in pypy programming.

# Enter your code here. Read input from STDIN. Print output to STDOUT
import numpy
n, m = map(int, raw_input().strip().split())
A = []
B = []
for _ in range(n):
    A.append(map(int, raw_input().strip().split()))
for _ in range(n):
    B.append(map(int, raw_input().strip().split()))
    
numpy.array(A)
numpy.array(B)

print numpy.add(A, B)
print numpy.subtract(A, B)
print numpy.multiply(A, B)
print numpy.divide(A, B)
print numpy.mod(A, B)
print numpy.power(A, B)

Problem solution in pypy3 programming.

# Enter your code here. Read input from STDIN. Print output to STDOUT
import numpy
n, m = map(int, raw_input().strip().split())
A = []
B = []
for _ in range(n):
    A.append(map(int, raw_input().strip().split()))
for _ in range(n):
    B.append(map(int, raw_input().strip().split()))
    
numpy.array(A)
numpy.array(B)

print numpy.add(A, B)
print numpy.subtract(A, B)
print numpy.multiply(A, B)
print numpy.divide(A, B)
print numpy.mod(A, B)
print numpy.power(A, B)