Python返回最常见的元素

Question

我正在尝试创建一个函数，该函数从传递的数组中返回最常见的元素。我的代码如下：

def get_classification(classes):
    from collections import Counter
    count = Counter(classes)
    return count.most_common()[0]

它正确地返回了最常见的元素。但是，它以element, count格式返回，例如：

3.0, 2
2.0, 3
1.0, 3

我不希望它返回该元组。我 just 需要它来返回最常见的元素。那怎么可能？

我尝试了解决方法：

counts = numpy.bincount(classes)
final = numpy.argmax(counts)

return final

但这对我也不起作用。 任何建议将不胜感激。

Answer 1

您可以再次使用索引，以获取元组的第一个元素：

def get_classification(classes):
    from collections import Counter
    count = Counter(classes)
    return count.most_common()[0][0]

print(get_classification([1, 3, 3, 1, 2, 1])) # ==> 1