子集Numpy数组,其中值在指定范围内

nuypyhwy  于 11个月前  发布在  其他
关注(0)|答案(1)|浏览(85)

给定以下数组,我如何返回一个新数组,该数组包含指定范围内每组值的第一个值。可能有任何大于2的值。
Eg

specified_range = 10

in_array = np.array([3, 14, 25, 37, 38, 50, 61, 66, 70])

字符串
指定范围内的值组:[37,38]和[61,66,70]
预期结果:[37 61]

3phpmpom

3phpmpom1#

如果数组是按升序排序的,则使用以下代码:

s = np.diff(in_array) < specified_range
in_array[:-1][s & (np.r_[False, s[:-1]] != s)]

array([37, 61])

字符串

相关问题