Hacker rank Compare the Triplets

#!/bin/python3
 
import math
import os
import random
import re
import sys
 
#
# Complete the 'compareTriplets' function below.
#
# The function is expected to return an INTEGER_ARRAY.
# The function accepts following parameters:
#  1. INTEGER_ARRAY a
#  2. INTEGER_ARRAY b
#
 
def compareTriplets(a, b):
    # Write your code here
    cntA,cntB=0,0
    lenArr=len(a)
    for i in range (lenArr):
        if a[i]>b[i] : cntA+=1
        if a[i]<b[i] : cntB+=1
    return cntA,cntB
 
if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')
 
    a = list(map(int, input().rstrip().split()))
 
    b = list(map(int, input().rstrip().split()))
 
    result = compareTriplets(a, b)
 
    fptr.write(' '.join(map(str, result)))
    fptr.write('\n')
 
    fptr.close()