python小练习(二)

题目：
Py从小喜欢奇特的东西，而且天生对数字特别敏感，一次偶然的机会，他发现了一个有趣的四位数2992， 这个数，它的十进制数表示，其四位数字之和为2+9+9+2=22，它的十六进制数BB0，其四位数字之和也为22， 同时它的十二进制数表示1894，其四位数字之和也为22，啊哈，真是巧啊。 Py非常喜欢这种四位数，由于他的发现，所以这里我们命名其为Py数。 现在给你一个十进制4位数n，你来判断n是不是Py数，若是，则输出Yes，否则输出No。 如n=2992，则输出Yes； n = 9999，则输出No。

思路：把不同进制下各个位置上的数取出来相加既可

# 取不同进制数的各个位置上的数并求和
def cal_function(num,times):
    temp = 0
    answer = 0
    for i in range(len(str(len))):
        temp = num % times
        num //= times
        answer += temp
    return answer
n = int(input("请输入一个数："))
if (cal_function(n, 10) == cal_function(n, 12)) and (cal_function(n, 12) == cal_function(n, 16)) \
        and (cal_function(n, 10) == cal_function(n, 16)):
    print('Yes')
else:
    print('No')

如果有更简洁的写法，欢迎各位朋友在评论区告诉我哦，谢谢~~