Python代碼找出所有水仙花數(shù)的方法:首先用列表推到遍歷所有的三位數(shù),每個(gè)數(shù)都打上標(biāo)記;然后篩選出標(biāo)記為T(mén)rue的元組;最后第一句篩選出來(lái)的元組的第二個(gè)值放到list結(jié)構(gòu)里,并加上print語(yǔ)句即可。
Python代碼找出所有水仙花數(shù)的方法:
方法一:
>>> >>> 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] >>>
相關(guān)學(xué)習(xí)推薦:python視頻教程
說(shuō)明:
上面的代碼可以分解三句代碼:
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))
第一句表示用列表推到遍歷所有的三位數(shù),每個(gè)數(shù)都打上標(biāo)記,是水仙花數(shù)著標(biāo)記True,不是則標(biāo)記False,標(biāo)記和數(shù)放到一個(gè)元祖:(flag, value),所有元組放在一個(gè)list結(jié)構(gòu)里。
第二句表示篩選出標(biāo)記為T(mén)rue的元組。
第三句表示把c。
加上print語(yǔ)句,執(zhí)行一遍上面三句代碼即可明白。
方法二:
只用列表推導(dǎo)。
>>> 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] >>>