Python學習路線實用技法：篩選序列中的元素

網(wǎng)站設(shè)計制作過程拒絕使用模板建站；使用PHP+MYSQL原生開發(fā)可交付網(wǎng)站源代碼;符合網(wǎng)站優(yōu)化排名的后臺管理系統(tǒng)；做網(wǎng)站、成都網(wǎng)站建設(shè)收費合理；免費進行網(wǎng)站備案等企業(yè)網(wǎng)站建設(shè)一條龍服務.我們是一家持續(xù)穩(wěn)定運營了10余年的成都創(chuàng)新互聯(lián)網(wǎng)站建設(shè)公司。

序列中含有一些數(shù)據(jù)，我們需要提取其中的值或根據(jù)某些標準對序列做刪減，

要篩選序列中的數(shù)據(jù)，通常最簡單的方法是使用列表推導式。

例如：

myList=[1,4,-5,10,-7,2,3,-1]
print([n for n in myList if n>0])
print([n for n in myList if n<0])

結(jié)果：

[1, 4, 10, 2, 3]
[-5, -7, -1]

使用列表推導式的一個潛在缺點是如果原始輸入非常大的話，這么做可能會產(chǎn)生一個龐大的結(jié)果。如果這是你需要考慮的問題，那么可以使用生成器表達式通過迭代的方法產(chǎn)生篩選結(jié)果，例如：

myList=[1,4,-5,10,-7,2,3,-1]
pos=(n for n in myList if n >0)
for x in pos:
 print(x)

結(jié)果：

1
4
10
2
3

有時候篩選的標準沒法簡單地表示在列表推導式或生成器表達式中。比如：假設(shè)篩選過程涉及異常處理或者其他一些復雜的細節(jié)?？梢詫⑻幚砗Y選邏輯的代碼放到單獨的函數(shù)中，然后使用內(nèi)建的filter()函數(shù)處理，示例如下：

values=['1','2','-3','-','4','N/A','5']
def is_int(val):
 try:
 x=int(val)
 return True
 except ValueError:
 return False
ivals=list(filter(is_int,values))
print(ivals)

結(jié)果：

['1', '2', '-3', '4', '5']

filter()創(chuàng)建了一個迭代器，因此如果我們想要的是列表形式的結(jié)果，請確保加上了list()，就像示例中那樣。

分析

列表推導式和生成器表達式通常是用來篩選數(shù)據(jù)的最簡單和最直接的方式。此外，它們也具有同時對數(shù)據(jù)做轉(zhuǎn)換的能力。例如：

import math
myList=[1,4,-5,10,-7,2,3,-1]
print([math.sqrt(n) for n in myList if n>0])

結(jié)果：

[1.0, 2.0, 3.1622776601683795, 1.4142135623730951, 1.7320508075688772]

關(guān)于篩選數(shù)據(jù)，有一種情況是用新值替換掉不滿足標準的值，而不是拋棄它們。例如。除了要找到正整數(shù)之外，我們也希望在指定范圍內(nèi)將不滿足要求的值替換掉。通常，這可以通過將篩選條件遷移到一個條件表達式中來輕松實現(xiàn)，就像下面這樣：

myList=[1,4,-5,10,-7,2,3,-1]
print([n if n>0 else 0 for n in myList])
print([n if n<0 else 0 for n in myList])

結(jié)果：

[1, 4, 0, 10, 0, 2, 3, 0]
[0, 0, -5, 0, -7, 0, 0, -1]

另一個值得一提的篩選工具是itertools.compress()，它接受一個可迭代對象以及一個布爾選擇器序列作為輸入。輸出時，它會給出所有在相應的布爾選擇器中為True的可迭代對象元素。如果想把對一個序列的篩選結(jié)果施加到另一個相關(guān)的序列上時，這就會非常有用。

例如：

from itertools import compress
address=[
'5412 N CLARK1',
'5148 N CLARK2',
'5800 E CLARK3',
'2122 N CLARK4',
'5645 M CLARK5',
'1060 W CLARK6',
]
counts=[0,3,10,4,1,7]
#構(gòu)建一個列表，它相應的count值要大于5
more5=[n>5 for n in counts]
print(more5)
print(list(compress(address,more5)))

結(jié)果：

[False, False, True, False, False, True]
['5800 E CLARK3', '1060 W CLARK6']

這里的關(guān)鍵在于首先創(chuàng)建一個布爾序列，用來表示哪個元素可滿足我們的條件，然后compress()函數(shù)挑選出滿足布爾值為True的相應元素。

同filter()函數(shù)一樣，正常情況下，compress()會返回一個迭代器。因此，如果需要的話，得使用list()將結(jié)果轉(zhuǎn)為列表。