擅长:python、mysql、java
<p>我相信您应该可以在这里通过使用<code>return_inverse</code>在<code>np.unique()</code>内使用<code>return_inverse</code>:</p>
<blockquote>
<p>If True, also return the indices of the unique array (for the
specified axis, if provided) that can be used to reconstruct <code>ar</code>.</p>
</blockquote>
<pre><code>>>> import numpy as np
>>> a = np.array([[2,2,3,3],
... [2,3,3,3],
... [3,3,4,4]])
>>> _, inv, cts = np.unique(a, return_inverse=True, return_counts=True)
>>> cts[inv].reshape(a.shape)
array([[3, 3, 7, 7],
[3, 7, 7, 7],
[7, 7, 2, 2]])
</code></pre>
<p>这也适用于扁平数组未排序的情况,例如<code>b = np.array([[1, 2, 4], [4, 4, 1]])</code>。在</p>