快捷導(dǎo)航

np.unique()的具體使用

更新時(shí)間：2023年03月14日 10:38:33 作者：想變厲害的大白菜

本文主要介紹了np.unique()的具體使用，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

一、np.unique() 介紹

對(duì)于一維數(shù)組或者列表，np.unique() 函數(shù) 去除其中重復(fù)的元素，并按元素由小到大返回一個(gè)新的無元素重復(fù)的元組或者列表。

二、np.unique() 原型

numpy.unique(arr, return_index, return_inverse, return_counts)

arr：輸入數(shù)組，如果不是一維數(shù)組則會(huì)展開
return_index：如果為 true，返回新列表元素在舊列表中的位置（下標(biāo)），并以列表形式存儲(chǔ)。
return_inverse：如果為true，返回舊列表元素在新列表中的位置（下標(biāo)），并以列表形式存儲(chǔ)。
return_counts：如果為 true，返回去重?cái)?shù)組中的元素在原數(shù)組中的出現(xiàn)次數(shù)。

三、實(shí)例

import numpy as np
A = [1, 2, 2, 5, 3, 4, 3]
a = np.unique(A)
print(a)
print("______")

a, indices = np.unique(A, return_index=True) ? # 返回新列表元素在舊列表中的位置（下標(biāo)）
print(a)?? ??? ? # 列表
print(indices)?? ? # 下標(biāo)
print("______")

a, indices = np.unique(A, return_inverse=True) ? # 舊列表的元素在新列表的位置
print(a)
print(indices)
print(a[indices]) ? ? # 使用下標(biāo)重構(gòu)原數(shù)組
print("______")

a, indices = np.unique(A, return_counts=True) ? ?# 每個(gè)元素在舊列表里各自出現(xiàn)了幾次
print(a)
print(indices)
print("______")

B = ([1, 2], [2, 5], [3, 4])
b = np.unique(B)
C= ['fgfh','asd','fgfh','asdfds','wrh']
c= np.unique(C)
print(b)
print(c)

輸出結(jié)果：

[1 2 3 4 5]
______
[1 2 3 4 5]
[0 1 4 5 3]
______
[1 2 3 4 5]
[0 1 1 4 2 3 2]
[1 2 2 5 3 4 3]
______
[1 2 3 4 5]
[1 2 2 1 1]
______
[1 2 3 4 5]
['asd' 'asdfds' 'fgfh' 'wrh']