我有两个形状相同的二维numpy数组。有没有一种同时遍历它们并获取来自两个数组的一对元素及其索引的方法?
例如,我有两个数组:
before = np.array(
[[0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0]],
dtype=int
)
after = np.array(
[[0, 0, 1, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 1, 0, 0, 1]],
dtype=int
)
我想获取一个列表,其中包含在“after”表中转换为“one”的每个“zero”在“before”表中的索引 - 在此场景中将会是
[(0, 2), (1, 4), (1, 7)]
。
numpy.ndenumerate
非常接近我想要实现的功能,但它只能一次迭代一个数组。