計算 NumPy 陣列中的唯一值
Muhammad Maisam Abbas
2021年10月2日
2021年7月4日
本教程將介紹如何計算 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()
函式將兩個一維陣列 unique
和 counts
壓縮到單個二維陣列 result
中。
Author: Muhammad Maisam Abbas
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