擅长:python、mysql、java
<p>@Bas的答案可能就是你要找的。但这里有另一种方法,使用numpy的<code>vectorize</code>技巧:</p>
<pre><code>import numpy as np
S = set([0,2,6,8])
@np.vectorize
def contained(x):
return x in S
contained(fake)
=> array([[ True, False, True],
[False, False, False],
[ True, False, True]], dtype=bool)
</code></pre>
<p>这个解决方案的优点是为每个元素(即在python空间中)调用<code>contained()</code>,这使得这个过程比纯numpy解决方案慢得多。</p>