Маскировать определенные значения из массива
Пример:
У меня есть массив:
array([[1, 2, 0, 3, 4],
[0, 4, 2, 1, 3],
[4, 3, 2, 0, 1],
[4, 2, 3, 0, 1],
[1, 0, 2, 3, 4],
[4, 3, 2, 0, 1]], dtype=int64)
У меня есть набор (переменная длина, порядок не имеет значения) "плохих" значений:
{2, 3}
Я хочу вернуть маску, которая скрывает эти значения:
array([[False, True, False, True, False],
[False, False, True, False, True],
[False, True, True, False, False],
[False, True, True, False, False],
[False, False, True, True, False],
[False, True, True, False, False]], dtype=bool)
Какой самый простой способ сделать это в NumPy?