【python練習】函數

1.編程實現 9*9乘法表(循環嵌套的複習)




2.平方等式

def f(n):
    n = str(n)
    sum = 0
    for item in n:
        sum += int(item) ** 2
    return sum


def isOK(k, n):
    if k * f(n) == n:
        return True
    else:
        return False


def main():
    k = 51
    a = 5000
    b = 10000
    count = 0
    for i in range(a, b + 1):
        if isOK(k, i):
            count += 1
    print(count)

main()

執行效果：