如何使用numpy python计算元素向量的数量

例如，如果我有：

a=np.array([[1,1,4,1,4,3,1]])

我们可以看到，数字 1 出现了四次，数字 4 出现了两次，只有 3 次。

我想要得到以下结果：

array(4,4,2,4,2,1,4)

正如您所看到的：每个单元格都被其元素的计数替换。

我怎样才能以最有效的方式做到这一点？

One vectorized接近与np.unique and np.searchsorted -

# Get unique elements and their counts
unq,counts = np.unique(a,return_counts=True)

# Get the positions of unique elements in a. 
# Use those positions to index into counts array for final output.
out = counts[np.searchsorted(unq,a.ravel())]

样本运行 -

In [86]: a
Out[86]: array([[1, 1, 4, 1, 4, 3, 1]])

In [87]: out
Out[87]: array([4, 4, 2, 4, 2, 1, 4])

根据@Jaime 的评论，您可以使用np.unique像这样独自一人——

_, inv_idx, counts = np.unique(a, return_inverse=True, return_counts=True)
out = counts[inv_idx]