我有两个numpy数组(my_numpy_array_1
和my_numpy_array_2
)。
我想遍历两者,找出相应的索引和值。
我的问题是,如果my_numpy_array_2
大于0,则需要在idx_data_2
上进行迭代时设置起始索引。
如何枚举范围,如下面的代码(但是下面的代码示例不起作用)?
idx = 0
for idx_data_1, val_data_1 in np.ndenumerate(my_numpy_array_1):
# How can I do this using a new start index (idx), when re-entering for loop?
idx_data_2, val_data_2 = (min(enumerate(range (idx, my_numpy_array_2.size, 1),
key=lambda x: abs(x[1] - val_data_1)))
if idx_data_2 > 0:
idx = idx_data_2 + 1