Как я могу реализовать команду `` ismember () `` matlabs в Python?
вот моя проблема: я хотел бы создать булеву матрицу B, которая содержитTrue
везде, где матрица A имеет значение, содержащееся в векторе v. Одним из неудобных решений будет:
import numpy as np
>>> A = np.array([[0,1,2], [1,2,3], [2,3,4]])
array([[0, 1, 2],
[1, 2, 3],
[2, 3, 4]])
>>> v = [1,2]
>>> B = (A==v[0]) + (A==v[1]) # matlab: ``B = ismember(A,v)``
array([[False, True, True],
[ True, True, False],
[ True, False, False]], dtype=bool)
Может быть, есть решение, которое было бы более удобным, если бы у A и v было больше значений?
Ура!