python3中如何在清單，字典，集合中根據條件刷選資料

實際案例

通用做法是：疊代

data = [1, 2,4,5,6]
res = []
for x in data:
    if x>=6:
    res.append(x
      

解決方案

filter簡介

Python 内建的 filter()函數用于過濾序列。 和 map()類似， filter()也接收一個函數和一個序列。和 map()不同的時， filter()把傳入的函數依次作用于每個元素，然後根據傳回值是 True 還 是 False 決定保留還是丢棄該元素。

In [4]:

def is_odd(n):
    return n % 2 == 1
      

注意到 filter()函數傳回的是一個 Iterator，也就是一個惰性序列，所 以要強迫 filter()完成計算結果，

需要用 list()函數獲得所有結果并返 回 list。

In [12]:

a = (filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
list(a)
      

Out[12]:

[1, 5, 9, 15]      

In [ ]: In [6]:

from random import randint
# Python中對于無需關注其實際含義的變量可以用_代替，
# 這就和for  i in range(5)一樣，因為這裡我們對i并不關心，是以用_代替僅擷取值而已。
data = [randint(-10, 10) for _ in range(10)]
data
      

Out[6]:

[-1, -5, -7, 10, -7, -5, -6, 8, 6, -8]      

In [8]:

匿名函數有個限制，就是隻能有一個表達式，不用寫 return，傳回值就是該表達式的結果。

用匿名函數有個好處，因為函數沒有名字，不必擔心函數名沖突。此外， 匿名函數也是一個函數對象，也可以把匿名函數指派給一個變量，再利 用變量來調用該函數： 同樣，也可以把匿名函數作為傳回值傳回，比如：

def build(x, y):
    return lambda: x * x + y * y
      

Python 對匿名函數的支援有限，隻有一些簡單的情況下可以使用匿名函 數。

In [16]: Out[16]:

[10, 8, 6]      

In [15]: Out[15]:

[-1, -5, -7, 10, -7, -5, -6, 8, 6, -8]      

清單解析

In [17]: Out[17]:

[10, 8, 6]      

In [21]:

%timeit list(filter(lambda x: x>=0, data))
      

1.9 µs ± 19.9 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
      

In [22]:

687 ns ± 3.89 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
      

In [23]: In [24]: Out[24]:

{1: 100,
 2: 65,
 3: 67,
 4: 77,
 5: 71,
 6: 92,
 7: 68,
 8: 75,
 9: 97,
 10: 92,
 11: 77,
 12: 96,
 13: 75,
 14: 95,
 15: 66,
 16: 67,
 17: 92,
 18: 77,
 19: 90,
 20: 80}      

字典解析

預設情況下， dict 疊代的是 key。如果要疊代 value，可以用 for value in d.values()，

如果要同時疊代 key 和 value，可以用 for k, v in d.items()。

In [32]: Out[32]:

{1: 100, 6: 92, 9: 97, 10: 92, 12: 96, 14: 95, 17: 92, 19: 90}      

In [33]: Out[33]:

[-1, -5, -7, 10, -7, -5, -6, 8, 6, -8]      

In [34]: In [35]: Out[35]:

{-8, -7, -6, -5, -1, 6, 8, 10}      

集合解析

In [36]:

{x for x in s if x%3 ==0}
      

Out[36]:

{-6, 6}