Python 盈数和亏数的个数 问题如图

import math

def num_type(n):
    s = 1
    for i in range(2, int(math.sqrt(n)) + 1):
        if n % i == 0:
            s += i
            j = n // i
            if j != i:
                s += j
    if s < n:
        return -1
    elif s > n:
        return 1
    else:
        return 0

n, m = [int(x) for x in input().split(" ")]
counts = [0, 0, 0]
for i in range(n, m):
    result = num_type(i)
    counts[result + 1] += 1
print(counts)