np.unique

import numpy as np

A = [1, 2, 2, 5,3, 4, 3]

a = np.unique(A)

B= (1, 2, 2,5, 3, 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']

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

print(a)

print(s)

# 运行结果

# [1 2 3 4 5]

# [0 1 4 5 3]

a, s,p = np.unique(A, return_index=True, 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]