Interseção da matriz Numpy 2d e 1d
Para cada elemento na matrizA[:,3:]
que também está na matrizB
, Desejo definir o valor como 0, o que cria a 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]])
Eu sei que existe uma maneira de fazer isso usandoin1d
e radiodifusãoA
como uma matriz 1D, mas não tenho idéia de como fazer isso.
Qualquer ajuda seria muito apreciada.