np.unique

a = np.unique(A)

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

代码语言：javascript复制

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']

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

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

代码语言：javascript复制

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

print(a)

print(s)

# 运行结果

# [1 2 3 4 5]

# [0 1 4 5 3]

代码语言：javascript复制

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

代码语言：javascript复制

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

代码语言：javascript复制

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]

还少一个，return_counts = true,返回个数（用于统计各个元素出现的次数）

return unique 函数 数组 统计

0 人点赞