python二维数组去重_np.unique()对一维和二维数组去重

一维数组

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

import numpy as np

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

a = np.unique(A)

print(a)

[1 2 3 4 5]

C= ['fgfh','asd','fgfh','asdfds','wrh']

c= np.unique(C)

print(c)

['asd' 'asdfds' 'fgfh' 'wrh']

return_index = True

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

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

print(a)

print(s)

[1 2 3 4 5]

[0 1 4 5 3]

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