溫馨提示×

numpy怎么篩選符合條件的行

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

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

布爾索引是一種使用布爾數組來選擇數組中的元素的方法。我們可以通過創建一個布爾數組來指定篩選條件,然后將其與原數組進行索引,以獲取符合條件的行。

以下是一個例子:

import numpy as np

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

# 創建一個布爾數組,指定篩選條件
condition = arr > 5

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

print(filtered_arr)

輸出結果為:

[6 7 8 9]

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

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女