Пересечение 2d и 1d Numpy массива
Для каждого элемента в массивеA[:,3:]
это также в массивеB
Я хочу установить значение 0, который создает массивresult
import numpy as np
A = np.array([[1, 1, 10, 101, 102, 103, 0, 0],
[2, 2, 10, 102, 108, 0, 0, 0],
[3, 3, 11, 101, 102, 106, 107, 108]])
B = np.array([101, 106, 108])
result = np.array([[1, 1, 10, 0, 102, 103, 0, 0],
[2, 2, 10, 102, 0, 0, 0, 0],
[3, 3, 11, 0, 102, 0, 107, 0]])
Я знаю, что есть способ сделать это, используяin1d
и вещаниеA
как одномерный массив, но я понятия не имею, как это сделать.
Любая помощь будет принята с благодарностью.