过滤二维 numpy 数组

Question

我有一个形状为 (2,50,000) 的 2D numpy 数组 - 这意味着我有 50k 个 x,y 值样本。

我想过滤某个范围内的 x,y 值，比如：

min < x,y < max

我尝试将 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]