python自学做题记录之py数

题目描述：
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。
示例：
输入：n = 1234

输出：No

这题如果是二进制，八进制，十六进制问题就好解决，直接使用内置函数bin,oct,hex转换就行，但是这里有一个十二进制，所以说可以用以下代码

def get_system(num, s):
    sums = 0
    
    while num:
        sums += num % s
        num //= s
    return sums
if get_system(n, 10) == 22 and get_system(n, 16) == 22 and get_system(n, 12) == 22:
    print("Yes")
else:
    print("No")