擅长:python、mysql、java
<p>对于NumPy数组,您可能希望以矢量化的方式工作以提高性能,并使用数组切片。有鉴于此,这里有一种输入数组的方法<code>a</code>和<code>b</code>-</p>
<pre><code>n = min(len(a), len(b))
out_idx = np.flatnonzero(a[:n] == b[:n])
out_val = a[out_idx] # or b[out_idx] both work
</code></pre>
<p>这会处理多个匹配项。</p>
<p>样本运行-</p>
<pre><code>In [224]: a = np.array([3, 8, 9, 2, 1, 7])
In [225]: b = np.array([1, 2, 9, 7, 5, 7, 0, 4])
In [226]: n = min(len(a), len(b))
...: out_idx = np.flatnonzero(a[:n] == b[:n])
...: out_val = a[out_idx]
...:
In [227]: out_idx
Out[227]: array([2, 5])
In [228]: out_val
Out[228]: array([9, 7])
</code></pre>
<p>对于作为索引输出的元组列表及其值-</p>
<pre><code>In [229]: zip(out_idx, out_val)
Out[229]: [(2, 9), (5, 7)]
</code></pre>
<p>对于索引和相应值的漂亮字典输出-</p>
<pre><code>In [230]: {i:j for i,j in zip(out_idx, out_val)}
Out[230]: {2: 9, 5: 7}
</code></pre>