首頁 >後端開發 >Python教學 >Python如何篩選序列中的元素

Python如何篩選序列中的元素

不言
不言轉載
2018-10-22 17:20:482499瀏覽

這篇文章帶給大家的內容是關於Python如何篩選序列中的元素 ,有一定的參考價值,有需要的朋友可以參考一下,希望對你有所幫助。

1、需求

序列中含有一些數據,我們需要提取其中的值或根據某些標準對序列做刪減,

#2、解決方案

要篩選序列中的數據,通常最簡單的方法是使用清單推導式。

例如:

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])

結果:

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

使用列表推導式的一個潛在缺點是如果原始輸入非常大的話,這麼做可能會產生一個龐大的結果。如果這是你需要考慮的問題,那麼可以使用生成器表達式透過迭代的方法產生篩選結果,例如:

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

結果:

1
4
10
2
3

有時候篩選的標準沒辦法簡單地表示在清單推導式或產生器表達式中。例如:假設篩選過程涉及異常處理或其他一些複雜的細節。基於從,可以將處理篩選邏輯的程式碼放到單獨的函數中,然後使用內建的filter()函數處理,範例如下:

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)

結果:

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

filter()創建了一個迭代器,因此如果我們想要的是列表形式的結果,請確保加上了list(),就像範例一樣。

3、分析

列表推導式和生成器表達式通常是用來篩選資料的最簡單和最直接的方式。此外,它們也具有同時對資料做轉換的能力。例如:

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

結果:

[1.0, 2.0, 3.1622776601683795, 1.4142135623730951, 1.7320508075688772]

關於篩選數據,有一種情況是用新值替換掉不符合標準的值,而不是拋棄它們。例如。除了要找到正整數之外,我們也希望在指定範圍內將不符合要求的值替換掉。通常,這可以透過將篩選條件遷移到一個條件表達式中來輕鬆實現,就像下面這樣:

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])

結果:

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

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

例如:

from itertools import compress
address=[
&#39;5412 N CLARK1&#39;,
&#39;5148 N CLARK2&#39;,
&#39;5800 E CLARK3&#39;,
&#39;2122 N CLARK4&#39;,
&#39;5645 M CLARK5&#39;,
&#39;1060 W CLARK6&#39;,
]
counts=[0,3,10,4,1,7]
#构建一个列表,它相应的count值要大于5
more5=[n>5 for n in counts]
print(more5)

print(list(compress(address,more5)))

結果:

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

這裡的關鍵在於先建立一個布林序列,用來表示哪個元素可滿足我們的條件,然後compress()函數挑選出滿足布林值為True的相應元素。

同filter()函數一樣,正常情況下,compress()會傳回一個迭代器。因此,如果需要的話,得使用list()將結果轉為列表。

#

以上是Python如何篩選序列中的元素的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:segmentfault.com。如有侵權,請聯絡admin@php.cn刪除