1. 程式人生 > >Python中numpy庫unique函式解析

Python中numpy庫unique函式解析

a = np.unique(A)對於一維陣列或者列表,unique函式去除其中重複的元素,並按元素由大到小返回一個新的無元素重複的元組或者列表
  1. import numpy as np  
  2. A = [1225,343]  
  3. a = np.unique(A)  
  4. B= (122,5343)  
  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中。

  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中

  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>