Python 求众数|极客教程

Python 求众数

什么是众数？

在统计学中，众数是指一组数据中出现次数最多的数值。在Python中，我们可以通过编写函数来求解一个数组中的众数。

方法一：使用collections.Counter

使用collections.Counter可以方便地统计一个数组中每个元素出现的次数，并返回一个字典，其中键为元素值，值为出现次数。我们可以通过对这个字典进行排序，找出出现次数最多的元素。

from collections import Counter

def majorityElement(nums):
    count = Counter(nums)
    return max(count, key=count.get)

nums = [1, 2, 2, 3, 4, 2, 2, 5, 2]
print(majorityElement(nums))

运行以上代码，我们可以得到输出为2，即数组nums中出现次数最多的元素为2。

方法二：使用numpy

另一种求解众数的方法是使用numpy库中的unique和argmax函数。

import numpy as np

def majorityElement(nums):
    unique, counts = np.unique(nums, return_counts=True)
    index = np.argmax(counts)
    return unique[index]

nums = [1, 2, 2, 3, 4, 2, 2, 5, 2]
print(majorityElement(nums))

同样地，运行以上代码，我们也可以得到输出为2。

方法三：使用字典

我们也可以使用字典来统计每个元素出现的次数，并找出出现次数最多的元素。

def majorityElement(nums):
    count = {}
    for num in nums:
        if num in count:
            count[num] += 1
        else:
            count[num] = 1
    return max(count, key=count.get)

nums = [1, 2, 2, 3, 4, 2, 2, 5, 2]
print(majorityElement(nums))

以上代码同样可以得到输出为2。