python - 如何在 Python 中实现 matlabs `` ismember( )`` 命令？

Question

这是我的问题:我想创建一个 bool 矩阵 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 有更多的值，是否有更方便的解决方案？

干杯!

Answer 1

我不太了解 numpy，这是一个原始的 python:

>>> A = [[0,1,2], [1,2,3], [2,3,4]]
>>> v = [1,2]
>>> B = [map(lambda val: val in v, a) for a in A]
>>>
>>> B
[[False, True, True], [True, True, False], [True, False, False]]

编辑:正如 Brooks Moses 指出的那样，一些简单的时间安排似乎表明，这个可能更好:

>>> B = [ [val in v for val in a] for a in A]