python 7-11 水仙花数

水仙花数是指一个N位正整数（N≥3），它的每个位上的数字的N次幂之和等于它本身。 例如：153=1×1×1+5×5×5+3×3×3。

本题要求编写程序,计算所有N位水仙花数。

输入格式:
输入在一行中给出一个正整数N（3≤N≤5）

输出格式:
按递增顺序输出所有N位水仙花数，每个数字占一行。

输入样例:
在这里给出一组输入。例如：

3

输出样例:
在这里给出相应的输出。例如：

153
370
371
407

def shui(n):

    for i in range(10**(n-1),10**n):
        m = n
        sum=0
        tem=i
        while m>0:
            sum+=(tem//(10**(m-1)))**n
            tem%=(10**(m-1))
            m-=1
        if sum==i:
            print(i)
a=int(input())
shui(a)