過濾二維 numpy 陣列

Question

我有一個形狀為 (2,50,000) 的 2D numpy 數組 - 這意味着我有 50k 個 x,y 值樣本。

我想過濾一定范圍內的 x,y 值，可以說：

min < x,y < max

我嘗試使用帶有過濾器np.apply_along_axis的 np.apply_along_axis，但我無法使其工作。

我很想看到一些執行這個簡單任務的pythonic方式！

Answer 1

當您的數組為arr且您的值為(_min, _max)時，請使用：

selection = np.logical_and(_min <= arr, arr <= _max)
selection = np.logical_and(selection[0], selection[1])
filtered_arr = arr[:, selection]