在Python中搜索两个3列的numpy数组并查找满足条件的位置

import numpy as np

array1 = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

array2 = np.array([[9, 8, 7],
                   [6, 5, 4],
                   [3, 2, 1]])

condition = np.where(np.sum(array1 + array2, axis=1) == 10)

print(condition)

import numpy as np

array1 = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

array2 = np.array([[9, 8, 7],
                   [6, 5, 4],
                   [3, 2, 1]])

condition = np.where(np.sum(array1 + array2, axis=1) == 10)
print(condition)