Python 數組過濾 – 優雅地對數組進行精準篩選

在Python編程中，經常需要篩選數組中的特定元素。

一、基本過濾方法

Python提供了基本的過濾方法，使用for循環遍曆數組，通過if語句篩選符合條件的元素。代碼如下：

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
result = []
for i in arr:
    if i % 2 == 0:
        result.append(i)
print(result)

輸出為：

[2, 4, 6, 8]

上述代碼中，arr是我們要篩選的數組，通過循環遍歷每一個元素，如果該元素滿足條件（被2整除），則將其加入到結果數組result中。

二、利用filter()方法

Python提供了filter()方法，用於篩選符合條件的元素。filter()方法接受一個函數和一個序列作為參數，函數返回值為布爾值。代碼如下：

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
result = filter(lambda x: x % 2 == 0, arr)
print(list(result))

輸出為：

[2, 4, 6, 8]

上述代碼中，lambda函數判斷當前元素是否被2整除，filter()方法將該函數應用於數組中的每個元素，返回值為True的元素被篩選出來，最終以列表的形式輸出。

三、利用列表推導式

Python提供了列表推導式，可以用一行代碼實現數組篩選。代碼如下：

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
result = [x for x in arr if x % 2 == 0]
print(result)

輸出為：

[2, 4, 6, 8]

上述代碼中，利用列表推導式將數組arr中滿足條件（被2整除）的元素篩選出來並存儲在result數組中。

四、高級過濾方法——numpy庫

numpy（Numerical Python）是Python的數值計算擴展庫，提供了強大的數組操作功能。在numpy中，可以通過比較運算符篩選符合條件的元素。代碼如下：

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
result = arr[arr % 2 == 0]
print(result)

輸出為：

[2 4 6 8]

上述代碼中，將數組轉換為numpy數組，通過比較運算符%篩選出滿足條件的元素，最終返回一個numpy數組。

總結

以上是Python中常見的幾種數組過濾方法，根據不同的需求選擇合適的方法可以使篩選更加高效。