Numpy库-unique_numpy unique-CSDN博客

本文链接：https://blog.csdn.net/Fwuyi/article/details/125010816

a = np.unique(A)

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

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

a是返回A去除其中重复的元素，并按元素由小到大排序的元组或者列表，b是返回该元素在list中第一次出现的索引的元组或者列表

a,b=np.unique(a, return_inverse=True)

a是返回A去除其中重复的元素，并按元素由小到大排序的元组或者列表，b是返回原list中每个元素在新的list中对应的索引

np.unique(a, return_counts=True)

a是返回A去除其中重复的元素，并按元素由小到大排序的元组或者列表，b是返回该元素在list中出现的次数

import numpy as np
a = [1, 5, 4, 2, 3, 3, 5]
print(np.unique(a))
# [1 2 3 4 5] array
print(np.unique(a, return_index=True))
# (array([1, 2, 3, 4, 5]), array([0, 3, 4, 2, 1]))
print(np.unique(a, return_inverse=True))
# (array([1, 2, 3, 4, 5]), array([0, 4, 3, 1, 2, 2, 4]))
print(np.unique(a, return_counts=True))
# (array([1, 2, 3, 4, 5]), array([1, 1, 2, 1, 2]))
p, q, m, n = np.unique(a, return_index=True, return_inverse=True, return_counts=True)
print(p, q, m, n)
# [1 2 3 4 5] [0 3 4 2 1] [0 4 3 1 2 2 4] [1 1 2 1 2]