計算 NumPy 陣列中的唯一值

Muhammad Maisam Abbas 2021年10月2日 2021年7月4日
計算 NumPy 陣列中的唯一值

本教程將介紹如何計算 NumPy 陣列中唯一值的出現次數。

使用 numpy.unique() 函式計算 NumPy 陣列中的唯一值

要計算 numpy 陣列中每個唯一元素的出現次數,我們可以使用 numpy.unique() 函式。它將陣列作為輸入引數,並按升序返回陣列內的所有唯一元素。我們可以將 return_counts 引數指定為 True 以獲取陣列中每個元素重複的次數。請參考以下程式碼示例。

import numpy as np

array = np.array([1,1,1,2,3,4,4,4])

unique, counts = np.unique(array, return_counts=True)

result = np.column_stack((unique, counts)) 
print (result)

輸出:

[[1 3]
 [2 1]
 [3 1]
 [4 3]]

我們首先使用 np.array() 函式建立了一個 NumPy 陣列。然後,我們使用 np.unique() 函式將 array 的所有唯一元素儲存在 unique 陣列中,並將它們各自的出現次數儲存在 counts 陣列中。然後,我們使用 np.column_stack() 函式將兩個一維陣列 uniquecounts 壓縮到單個二維陣列 result 中。

Muhammad Maisam Abbas avatar Muhammad Maisam Abbas avatar

Maisam is a highly skilled and motivated Data Scientist. He has over 4 years of experience with Python programming language. He loves solving complex problems and sharing his results on the internet.

LinkedIn