NumPy中与Pandas的value_counts函数等价的函数是什么？

Question

Pandas 的 value_counts 函数用于输出 Pandas 系列中每个元素的计数。

如果我需要计算列表中每个唯一元素的数量，则需要将其转换为 Pandas 系列，然后调用 value_counts 函数。

在 NumPy 中是否有类似的函数呢？

- user67275

Answer 1

np.unique 函数可以返回数组中唯一的元素，同时它还提供了一个可选参数 return_counts，使得 np.unique 返回所有唯一元素以及每个唯一元素在原始数组中出现的次数，这类似于 Pandas 中的 value_counts 函数：

>>> a = np.array([1,2,3,4,5,4,3,2,1])
>>> unique, counts = np.unique(a, return_counts=True)
>>> unique
array([1, 2, 3, 4, 5])

>>> counts
array([2, 2, 2, 2, 1])

- user17242583