水仙花數是指一個 3位正整數,它的每個位上的數字的 3 次冪之和等於它本身。(例如:1^3 + 5^3+ 3^3 = 153)
下面用一句代碼實現找出所有的水仙花數:
方法一:
>>>
>>> a = list(map(lambda x: x[1], filter(lambda x: x[0], [(i*100+j*10+k == i**3+j**3+k**3, i**3+j**3+k**3) for i in range(1, 10) for j in range(0, 10) for k in range(0, 10)])))
>>> print(a)
[153, 370, 371, 407]
>>>
說明:
上面的代碼可以分解三句代碼:
a = [(i*100+j*10+k == i**3+j**3+k**3, i**3+j**3+k**3) for i in range(1, 10) for j in range(0, 10) for k in range(0, 10)]
b = filter(lambda x: x[0], a)
c = list(map(lambda x: x[1], b))
第一句表示用列表推到遍歷所有的三位數,每個數都打上標記,是水仙花數着標記True,不是則標記False,標記和數放到一個元祖:(flag, value),所有元組放在一個list結構里。
第二句表示篩選出標記為True的元組。
第三句表示把第一句篩選出來的元組的第二個值放到list結構里。
加上print語句,執行一遍上面三句代碼即可明白。
方法二:
只用列表推導。
>>> a = [i**3+j**3+k**3 for i in range(1, 10) for j in range(0, 10) for k in range(0, 10) if i*100+j*10+k == i**3+j**3+k**3]
>>> print(a)
[153, 370, 371, 407]
>>>
非常感謝你的閱讀
————————————————
版權聲明:本文為CSDN博主「adrrry」的原創文章,遵循 CC 4.0 BY-SA 版權協議,轉載請附上原文出處鏈接及本聲明。
原文鏈接:https://blog.csdn.net/adrrry/article/details/105693586