Python程式碼如何找出所有水仙花數

2020-08-12 18:01:22

Python程式碼找出所有水仙花數的方法:首先用列表推到遍歷所有的三位數,每個數都打上標記;然後篩選出標記爲True的元組;最後第一句篩選出來的元組的第二個值放到list結構裡,並加上print語句即可。

Python程式碼找出所有水仙花數的方法:

方法一:

>>> 
>>> 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的元組。

第三句表示把c。

加上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]
>>>

相關推薦:

以上就是Python程式碼如何找出所有水仙花數的詳細內容,更多請關注php中文網其它相關文章!