python中找到第一个符合的数据点-python – 从满足条件的NumPy矩阵中的每一行中取N个第一个值...

方法#1

def takeN_le_per_row_broadcasting(a, b, N=3): # a, b : 1D, 2D arrays respectively

# First col indices in each row of b with <= corresponding one in a

idx = (b <= a[:,None]).argmax(1)

# Get all N ranged column indices

all_idx = idx[:,None] + np.arange(N)

# Finally advanced-index with those indices into b for desired output

return b[np.arange(len(all_idx))[:,None], all_idx]

方法#2

from skimage.util.shape import view_as_windows

def takeN_le_per_row_strides(a, b, N=3): # a, b : 1D, 2D arrays respectively

# First col indices in each row of b with <= corresponding one in a

idx = (b <= a[:,None]).argmax(1)

# Get 1D sliding windows for each element off data

w = view_as_windows(b, (1,N))[:,:,0]

# Use fancy/advanced indexing to select the required ones

return w[np.arange(len(idx)), idx]