回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我的问题是:我想创建一个布尔矩阵B,它包含<code>True</code>矩阵a在向量v中包含值的任何地方。一个不方便的解决方案是:</p>
<pre><code>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)
</code></pre>
<p>如果a和v有更多的值,是否有一个更方便的解决方案?在</p>
<p>干杯!在</p>