numpy.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None)[source]
一维数组
对一维数组或列表，unique()函数去除其中重复元素，并按元素大小返回一个新的无重复元组或列表。
import numpy as np
A = [1, 2, 2, 5,3, 4, 3]
a = np.unique(A)
print(a)
[1 2 3 4 5]
C= ['fgfh','asd','fgfh','asdfds','wrh']
c= np.unique(C)
print(c)
['asd' 'asdfds' 'fgfh' 'wrh']
return_index = True
# return_index=True表示返回新列表元素在旧列表中的位置，并以列表形式储存在s中。
a, s= np.unique(A, return_index=True)
print(a)
print(s)
[1 2 3 4 5]
[0 1 4 5 3]
return_inverse = True
# return_inverse=True 表示返回旧列表元素在新列表中的位置，并以列表形式储存在p中。
a, s,p = np.unique(A, return_index=True, return_inverse=True)
print(a)
print(s)
print(p)
[1 2 3 4 5]
[0 1 4 5 3]
[0 1 1 4 2 3 2]
二维数组
import numpy
c=np.array(((1,2),(3,4),(5,6),(7,8),(7,8),(3,4),(1,2)))
x=c[:,0]+c[:,1]*1j
print('转化为虚数:',x)
print('虚数去重后:',np.unique(x))
print(np.unique(x,return_index=True))#return_index:输出的元素索引值
idx=np.unique(x,return_index=True)[1]
print('二维数组去重:\n',c[idx])
转化为虚数: [ 1.+2.j 3.+4.j 5.+6.j 7.+8.j 7.+8.j 3.+4.j 1.+2.j]
虚数去重后: [ 1.+2.j 3.+4.j 5.+6.j 7.+8.j]
(array([ 1.+2.j, 3.+4.j, 5.+6.j, 7.+8.j]), array([0, 1, 2, 3], dtype=int64))
二维数组去重:
[[1 2]
[3 4]
[5 6]
[7 8]]
print('去重方案2:\n',np.array(list(set([tuple(t) for t in c]))))去重方案2:
[[1 2]
[3 4]
[5 6]
[7 8]]
来自：https://blog.csdn.net/yangyuwen_yang/article/details/79193770