首页 > 解决方案 > 枚举范围

问题描述

我有两个 numpy 数组(my_numpy_array_1my_numpy_array_2)。我想遍历两者,得到相应的索引和值。

我的问题是我需要在迭代时设置起始索引my_numpy_array_2,如果idx_data_2大于0。

如何枚举范围,如下面的代码(但下面的代码示例不起作用)?

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

标签: python-3.xnumpy

解决方案


推荐阅读