水仙花数的查找python

Question

水仙花数的查找。水仙花数是指一个三位数，它的每个位上的数字的三次幂之和等于它本身。例如：153是一个水仙花数，因为

Answer 1

题目没写完，不过水仙花数是个老题目了，

res=[]
for i in range(100,1000):
    a = str(i)
    if int(a[0])**3+int(a[1])**3+int(a[2])**3==i:
        res.append(i)
print(res)

结果

[153, 370, 371, 407]