擅长:python、mysql、java
<p>因为您使用的是<code>numpy</code>,所以使用更多的numpy函数可以大大加快速度。您当前在我的系统上的方法:</p>
<pre><code>%timeit v=[i for i in range(len(A)) if nd in A[i]]
1 loop, best of 3: 4.23 s per loop
</code></pre>
<p>相反,这要快40倍:</p>
<pre><code>%timeit v = np.where(np.sum((A == nd), axis=1) > 0)
10 loops, best of 3: 92.2 ms per loop
</code></pre>
<p>您还可以查看<code>np.in1d</code>,它类似于我上面使用的<code>A == nd</code>,但是可以与列表(类似于A==nd1或nd2或nd3)进行比较。你知道吗</p>