水仙花数:每个数字的三次方之和等于自身,如:153 = 1的三次方 + 5的三次方 + 3的三次方
1. 常规写法:
for i in range(100, 1000):
a = i % 10 # 个位数
b = i // 10 % 10 # 十位数
c = i // 100 # 百位数
if a ** 3 + b ** 3 + c ** 3 == i:
print(i)
结果是:
153
370
371
407
2. 一行代码的写法:
[i for i in range(100,1000) if sum(3**(i//10**j%10) for j in range(3))==i]
结果是:[153, 370, 371, 407]