python - 使用索引迭代两个 2D numpy 数组

Question

我有两个相同形状的 2D 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
)

我想从 before 表中获取每个 0 的索引列表，该表已转换为 after 中的 one 表 - 在这种情况下，它将是 [(0, 2), (1, 4), (1, 7)]。

numpy.ndenumerate 非常接近我想要实现的目标，但它一次只能遍历一个数组。

Answer 1

您可以将这两个条件传递给 np.logical_and，然后使用 np.argwhere 查找满足这两个条件的索引:

idx = np.argwhere(np.logical_and(before==0, after==1))

输出:

[[0 2]
 [1 4]
 [1 7]]