如何计算ndarray中元素的频率

时间:2013-09-10 14:57:01

标签: python numpy count element multidimensional-array

我有一个numpy ndarray字符串,想知道某个单词出现在数组中的频率。我发现了这个解决方案:

letters = numpy.array([["a","b"],["c","a"]])
print (numpy.count_nonzero(letters=="a"))
  

- →2

我只是想知道我是否解决了这个不必要的复杂问题,或者这是否是最简单的解决方案,因为对于列表,有一个简单的.count()。

1 个答案:

答案 0 :(得分:5)

您还可以使用sum

>>> letters = numpy.array([["a","b"],["c","a"]])
>>> (letters == 'a').sum()
2
>>> numpy.sum(letters == 'a')
2