Python中numpy库unique函数解析

时间:2025-03-30 20:07:00

a = (A)
对于一维数组或者列表,unique函数去除其中重复的元素,并按元素由大到小返回一个新的无元素重复的元组或者列表
import numpy as np
A = [1, 2, 2, 5,3, 4, 3]
a = (A)
B= (1, 2, 2,5, 3, 4, 3)
b= (B)
C= ['fgfh','asd','fgfh','asdfds','wrh']
c= (C)
print(a)
print(b)
print(c)
#   输出为 [1 2 3 4 5]
# [1 2 3 4 5]
# ['asd' 'asdfds' 'fgfh' 'wrh']

c,s=(b,return_index=True) 
return_index=True表示返回新列表元素在旧列表中的位置,并以列表形式储存在s中。
a, s= (A, return_index=True)
print(a)
print(s)
# 运行结果
# [1 2 3 4 5]
# [0 1 4 5 3]

a, s,p = (A, return_index=True, return_inverse=True)

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

a, s,p = (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]