Question

我有一个包含许多重复元素的数组，我想找到重复次数最多的非零元素的计数和值。我有点工作，但我不确定这是否是在python中这样做的正确方法。

一些测试代码：

import numpy as np

x = np.array([0, 0, 0, 10, 10, 10, 10, 10])

l = list()
for i in np.unique(x):
    l.extend((np.count_nonzero(x[x==i]),))

maximum_times = np.max(l)

这告诉我最大元素重复的次数，但它没有告诉我该元素是什么，也许使用for循环不是一个好主意，但我无法想出任何其他pythonic解决方案。

Answer 1

正如Ashwini所说，你可以使用return_counts=True。

x = np.array([0, 0, 0, 10, 10, 10, 10, 10])

elements, repeats = np.unique(x, return_counts=True)
index = repeats.argmax()
elem = elements[index]

print "Max elem is " + str(elem) + " with " + str(repeats[index]) + " repetitions."

numpy：查找数组中每个唯一元素的数量

1 个答案: