我有一个形状为 (2,50,000) 的 2D numpy 数组 - 这意味着我有 50k 个 x,y 值样本。
我想过滤某个范围内的 x,y 值,比如:
min < x,y < max
我尝试将 np.apply_along_axis
与过滤器功能一起使用,但我无法使其工作。
我很想看看执行这个简单任务的 Pythonic 方法!
答案 0 :(得分:0)
如果您的数组为 arr
,而您的值为 (_min, _max)
,请使用:
selection = np.logical_and(_min <= arr, arr <= _max)
selection = np.logical_and(selection[0], selection[1])
filtered_arr = arr[:, selection]