python：水仙花数的计算

水仙花数
水仙花数是指一个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

python代码：

n=int(input())
if n==3:
    for i in range(100,1000):
        a=int(i/100);b=int(i/10)%10;c=i%10
        if a*a*a+b*b*b+c*c*c==i:
            print(i)
if n==4:
    for i in range(1000,10000):
        a=int(i/1000);b=int(i/100)%10;c=int(i/10)%10;d=i%10
        if a*a*a*a+b*b*b*b+c*c*c*c+d*d*d*d==i:
            print(i)
if n==5:
    for i in range(10000,100000):
        a=int(i/10000);b=int(i/1000)%10;c=int(i/100)%10;d=int(i/10)%10;e=i%10
        if a*a*a*a*a+b*b*b*b*b+c*c*c*c*c+d*d*d*d*d+e*e*e*e*e==i:
            print(i)