Python中numpy库unique函数解析

a = np.unique(A)

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

[python]  view plain  copy

1. import numpy as np  
2. A = [1, 2, 2, 5,3, 4, 3]  
3. a = np.unique(A)  
4. B= (1, 2, 2,5, 3, 4, 3)  
5. b= np.unique(B)  
6. C= ['fgfh','asd','fgfh','asdfds','wrh']  
7. c= np.unique(C)  
8. print(a)  
9. print(b)  
10. print(c)  
11. #   输出为 [1 2 3 4 5]  
12. # [1 2 3 4 5]  
13. # ['asd' 'asdfds' 'fgfh' 'wrh']  

c,s=np.unique(b,return_index=True) 

return_index=True表示返回新列表元素在旧列表中的位置，并以列表形式储存在s中。

[python]  view plain  copy

1. <span style="font-size:18px;">a, s= np.unique(A, return_index=True)  
2. print(a)  
3. print(s)  
4. # 运行结果  
5. # [1 2 3 4 5]  
6. # [0 1 4 5 3]</span>  

a, s,p = np.unique(A, return_index=True, return_inverse=True)

return_inverse=True 表示返回旧列表元素在新列表中的位置，并以列表形式储存在p中

[python]  view plain  copy

1. <span style="font-size:18px;">a, s,p = np.unique(A, return_index=True, return_inverse=True)  
2. print(a)  
3. print(s)  
4. print(p)  
5. # 运行结果  
6. # [1 2 3 4 5]  
7. # [0 1 4 5 3]  
8. # [0 1 1 4 2 3 2]</span>