Nhập vào số nguyên dương N in ra màng hình số lượng ước của N vd: nhập N=6 in ra 4

Code

import math
def primeFactors(n):
    lst = []
    while n % 2 == 0:
        lst.append(2)
        n = n / 2
    for i in range(3,int(math.sqrt(n))+1,2):
        while n % i== 0:
            lst.append(i)
            n = n / i
    if n > 2:
        lst.append(n)
    return lst
n = int(input())
lst = primeFactors(n)
uoc = 1
for x in list(set(lst)):
    uoc *= (lst.count(x)+1)
print(uoc)