Сравните две матрицы и найдите соответствующий индекс в соответствующем столбце

#arrays #numpy #indexing

Вопрос:

Итак, у меня есть два массива. Для каждого элемента в столбце i в array_1 я хочу найти , существует ли один и тот же элемент в столбце i array_2 в. Если он существует, мне нужна соответствующая позиция/индекс в столбце i в array_2 . Если он не существует, верните значение False.

Вместо использования цикла for мне интересно, есть ли какая-либо векторизованная функция, которая эффективно выполняла бы эту задачу?

Ниже приведены два массива примеров и мой желаемый результат.

 array_1 = np.array([[1, 2], [2 , np.nan]])
array_2 = np.array([[2, np.nan], [1 , 3 ]])

My desired output:
np.array([[1, False], [0 , 0]])
 

Комментарии:

1. почему он возвращает False для » 2 » в позиции [0, 1]?

2. потому что 2 не отображается в столбце 2 array_2