numpy.unique()使用方法

2023-02-22 06:00:35

1. 引數說明

numpy.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None, *, equal_nan=True)

ar：輸入陣列，除非設定了下面介紹的axis引數，否則輸入陣列均會被自動扁平化成一個一維陣列。

return_index：(可選引數，布林型別)，如果為True則結果會同時返回被提取元素在原始陣列中的索引值(index)。

return_inverse：(可選引數，布林型別)，如果為True則結果會同時返回元素位於原始陣列的索引值(index)。

return_counts：(可選引數，布林型別)，如果為True則結果會同時每個元素在原始陣列中出現的次數。

axis：計算唯一性時的軸

返回值：返回一個排好序列的獨一無二的陣列。

2. 範例

2.1. 一維陣列

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

結果

array([1, 2, 3])

2.2. 二維陣列

a = np.array([[1, 0, 0], [1, 0, 0], [2, 3, 4]])
np.unique(a, axis=0)

結果

array([[1, 0, 0], [2, 3, 4]])

2.3. 返回索引

a = np.array(['a', 'b', 'b', 'c', 'a'])
u, indices = np.unique(a, return_index=True)

結果

array([0, 1, 3])
array(['a', 'b', 'c'], dtype='<U1')

2.4. 重建輸入矩陣

a = np.array([1, 2, 6, 4, 2, 3, 2])
u, indices = np.unique(a, return_inverse=True)
u[indices]

結果

array([1, 2, 3, 4, 6])
array([0, 1, 4, 3, 1, 2, 1])
array([1, 2, 6, 4, 2, 3, 2])

範例：嘗試用引數 return_counts 解決一個小問題。

# coding: utf-8
import numpy as np
 
# 任務: 統計 a 中元素個數, 找出出現次數最多的元素
a = np.array([1, 1, 1, 3, 3, 2, 2, 2, 2, 4, 5, 5])
 
# numpy.unique() 測試
b = np.unique(a)
print(b)
 
# 使用 return_counts=True 統計元素重複次數
b, count = np.unique(a, return_counts=True)
print(b, count)
 
# 使用 zip 將元素和其對應次數打包成一個個元組, 返回元組的列表
zipped = zip(b, count)
# for i, counts in zipped:
#     print("%d: %d" % (i, counts))  # 這裡列印zipped出來,
#                                    # 下面 max()會報
#                                    # ValueError: max() arg is an empty sequence
#                                    # 不知道為什麼 >_<
 
# 使用 max() 函數找出出現次數最多的元素
target = max(zipped, key=lambda x: x[1])
print(target)

參考文獻

numpy.unique()函數

numpy.unique — NumPy v1.24 Manual

到此這篇關於numpy.unique()使用方法的文章就介紹到這了,更多相關numpy.unique()內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！