Intersección de la matriz Numpy 2d y 1d
Para cada elemento en la matrizA[:,3:]
eso también está en matrizB
, Quiero establecer el valor en 0, lo que crea la matrizresult
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]])
Sé que hay una manera de hacer esto usandoin1d
y transmitiendoA
como una matriz 1D, pero no tengo idea de cómo hacerlo.
Cualquier ayuda sería muy apreciada.