Wie kann ich den matlabs-Befehl `` ismember () `` in Python implementieren?

Hier ist mein Problem: Ich möchte eine boolesche Matrix B erstellen, die enthältTrue Überall dort, wo die Matrix A einen Wert hat, der im Vektor v enthalten ist. Eine unbequeme Lösung wäre:

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)

Gibt es vielleicht eine Lösung, die bequemer wäre, wenn A und v mehr Werte hätten?

Prost!

Antworten auf die Frage(6)

Ihre Antwort auf die Frage