Python統計列表元素出現次數

一個拉風的名字發表於2018-02-10
Python

1. 引言

在使用Python的時候，通常會出現如下場景：

array =  [1, 2, 3, 3, 2, 1, 0, 2]

獲取array中元素的出現次數
複製程式碼

比如，上述列表中：0出現了1次，1出現了2次，2出現了3次，3出現了2次。

本文闡述了Python獲取元素出現次數的幾種方法。點選獲取完整程式碼。

2. 方法

獲取元素出現次數的方法較多，這裡我提出如下5個方法，謹供參考。下面的程式碼，傳入的引數均為 array = [1, 2, 3, 3, 2, 1, 0, 2]

2.1 Counter方法

該方法可以迅速獲取list中元素出現的次數，可以參考官方文件

from collections import Counter
def counter(arr):
    return Counter(arr).most_common(2) # 返回出現頻率最高的兩個數

# 結果：[(2, 3), (1, 2)]
複製程式碼

2.2 list中的count,獲取每個元素的出現次數

def single_list(arr, target):
    return arr.count(target)

# target=2，結果：3
複製程式碼

2.3 list中的count,獲取所有元素的出現次數

返回一個dict

def all_list(arr):
    result = {}
    for i in set(arr):
        result[i] = arr.count(i)
    return result

# 結果：{0: 1, 1: 2, 2: 3, 3: 2}
複製程式碼

2.4 Numpy花式索引，獲取每個元素的出現次數

def single_np(arr, target):
    arr = np.array(arr)
    mask = (arr == target)
    arr_new = arr[mask]
    return arr_new.size

# target=2，結果：3
複製程式碼

2.5 Numpy花式索引，獲取所有元素的出現次數

返回一個dict

def all_np(arr):
    arr = np.array(arr)
    key = np.unique(arr)
    result = {}
    for k in key:
        mask = (arr == k)
        arr_new = arr[mask]
        v = arr_new.size
        result[k] = v
    return result

# 結果：{0: 1, 1: 2, 2: 3, 3: 2}
複製程式碼

3. 總結

以上就是我總結的幾種Python獲取元素出現個數的方法。

值得一提的是，我所用的list所有元素都是整數

array = [1, 2, 3, 3, 2, 1, 0, 2]

如果list中包含其它型別的元素，比如

array = [1, 2, 3, 3, 2, 1, 'a', 'bc', 0.1]

這種情況下需要獲取 a或1 的出現次數時，2.4中函式的呼叫形式應當為：target='a' / target='1'

作者郵箱： mr.yxj@foxmail.com

轉載請告知作者，感謝

相關文章