如何计算 ndarray 中某个项目的出现次数？

Question

如何计算以下数组中 0 和 1 的数量？

 y = np.array([0, 0, 0, 1, 0, 1, 1, 0, 0, 0, 0, 1])

---

y.count(0) 给出：

numpy.ndarray 对象没有属性 count

原文由 mflowww 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

使用 numpy.unique ：

 import numpy
a = numpy.array([0, 3, 0, 1, 0, 1, 2, 1, 0, 0, 0, 0, 1, 3, 4])
unique, counts = numpy.unique(a, return_counts=True)

>>> dict(zip(unique, counts))
{0: 7, 1: 4, 2: 1, 3: 2, 4: 1}

使用 collections.Counter 的非 numpy 方法；

 import collections, numpy
a = numpy.array([0, 3, 0, 1, 0, 1, 2, 1, 0, 0, 0, 0, 1, 3, 4])
counter = collections.Counter(a)

>>> counter
Counter({0: 7, 1: 4, 3: 2, 2: 1, 4: 1})

原文由 Ozgur Vatansever 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

使用 numpy.count_nonzero 怎么样，比如

>>> import numpy as np
>>> y = np.array([1, 2, 2, 2, 2, 0, 2, 3, 3, 3, 0, 0, 2, 2, 0])

>>> np.count_nonzero(y == 1)
1
>>> np.count_nonzero(y == 2)
7
>>> np.count_nonzero(y == 3)
3

原文由 Aziz Alto 发布，翻译遵循 CC BY-SA 3.0 许可协议