【Python】np.unique() 函数介绍与使用

一、作用

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

二、参数

numpy.unique(arr, return_index, return_inverse, return_counts)

• arr：输入数组，如果不是一维数组则会展开
• return_index：如果为 true，返回新列表元素在旧列表中的位置（下标），并以列表形式存储。
• return_inverse：如果为true，返回旧列表元素在新列表中的位置（下标），并以列表形式存储。
• return_counts：如果为 true，返回去重数组中的元素在原数组中的出现次数。

三、示例

import numpy as np
A = [1, 2, 2, 5, 3, 4, 3]
a = np.unique(A)
print(a)

结果为：

(PyTorch) D:\CodeProject>D:/Anaconda/envs/PyTorch/python.exe d:/CodeProject/demo.py
[1 2 3 4 5]