溫馨提示×

numpy怎么篩選符合條件的行

小億
146
2024-01-25 09:41:29
欄目: 編程語言

可以使用numpy的布爾索引來篩選符合條件的行。

布爾索引是一種使用布爾數(shù)組來選擇數(shù)組中的元素的方法。我們可以通過創(chuàng)建一個(gè)布爾數(shù)組來指定篩選條件，然后將其與原數(shù)組進(jìn)行索引，以獲取符合條件的行。

以下是一個(gè)例子：

import numpy as np

# 創(chuàng)建一個(gè)示例數(shù)組
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 創(chuàng)建一個(gè)布爾數(shù)組，指定篩選條件
condition = arr > 5

# 使用布爾索引來篩選符合條件的行
filtered_arr = arr[condition]

print(filtered_arr)

輸出結(jié)果為：

[6 7 8 9]

在上面的例子中，我們創(chuàng)建了一個(gè)二維數(shù)組arr，并創(chuàng)建了一個(gè)布爾數(shù)組condition來指定篩選條件，即arr中的元素大于5。然后，我們使用布爾索引arr[condition]來獲得滿足條件的元素，并將其保存在filtered_arr中。最后，我們打印出filtered_arr，即符合條件的行。

0 踩