Python中numpy库的unique函数

unique函数
对于一维数组（元组）或列表，unique函数去除其中重复的元素，并按元素由小到大的顺序返回一个新的无重复元素列表

import numpy as np

A = [1, 2, 2, 5, 3, 4, 4]  #列表
a = np.unique(A)
B = (1, 2, 2, 5, 5, 4, 3)  #元组
b = np.unique(B)
C = ['fgfh','asd','fgfh','asdfds','wrh']
c = np.unique(C)
print(a)
print(b)
print(c)

‘’‘
输出
[1 2 3 4 5]
[1 2 3 4 5]
['asd' 'asdfds' 'fgfh' 'wrh']
’‘’

#设置return_index=True，表示再返回新列表的元素在原列表中的位置，并以列表形式储存在t中。
s, t = np.unique(A,return_index=True)
print(s)
print(t)

‘’‘
输出
[1 2 3 4 5]
[0 1 4 5 3]
’‘’

#设置return_inverse=True，表示再返回原列表元素在新列表中的位置，并以列表形式储存在z中
x, y, z = np.unique(A, return_index=True, return_inverse=True)
print(x)
print(y)
print(z)

‘’‘
输出
[1 2 3 4 5]
[0 1 4 5 3]
[0 1 1 4 2 3 3]
’‘’