1. 程式人生 > >python 列表推導 求100內的素數

python 列表推導 求100內的素數

列表推導式(list comprehension)是利用其他列表建立新列表(類似於數學術語中的集合推導式)的一種方法。它的工作方式類似於for迴圈,也很簡單:

In [39]: [x*x for x in range(10)]
Out[39]: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
比如一個字典如下:a={1:2,2:3,3:4},要求連結字典的鍵值,並且以“,”隔開,結果:“2,3,4”
程式碼如下:print ",".join([str(x) for x in a.keys()])

如果只想打印出那些能被3整除的平方數,只需要通過新增一個if部分在推導式中就可以完成:

In [41]: [x*x for x in xrange(10) if x % 3 == 0]
Out[41]: [0, 9, 36, 81]

  也可以增加更多的for語句的部分:

In [42]: [(x,y) for x in range(3) for y in range(3)]
Out[42]: [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

In [43]: [[x,y] for x in range(2) for y in range(2)]
Out[43]: [[0, 0], [0, 1], [1, 0], [1, 1]]