【Python基础】np.unique( )的用法

一穷二白到年薪百万

已于 2023-02-09 17:49:26 修改

阅读量2.5k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： Python学习文章标签： list

于 2022-04-22 11:00:58 首次发布

本文链接：https://blog.youkuaiyun.com/zfhsfdhdfajhsr/article/details/124340442

54 篇文章

订阅专栏

该博客介绍了numpy库中的unique函数，用于去除数组重复元素并排序。该函数不仅能简化数组，还支持返回元素首次出现的索引、元素在原数组中的逆序索引及计数。通过实例展示了基本用法，并详细解释了return_index、return_inverse和return_count等进阶参数的用途。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 功能简介

该函数是去除数组中的重复数字，并排序之后输出。这个函数的功能可以用集合操作实现，但是集合操作并不能够让输出结果按一定顺序排列（集合进行了哈希操作）。

import numpy as np

res = np.unique([1, 6, 7, 7, 8, 2, 3, 4, 4])
print(res)
# [1 2 3 4 6 7 8]

res2 = np.unique([1, 2, 3])
print(res2)
# [1 2 3]

除了上面的用法之外，np.unique()还有三个参数，下面一一介绍:

1.return_index：return_index为True时：会构建一个递增的唯一值的新列表，并返回新列表u中的元素在之前定义的旧列表arr中第一次出现值的索引indices
return_inverse：return_inverse为True时：会构建一个递增的唯一值的新列表，并返回旧列表arr中的值在新列表u中的索引列表indices
return_count为True时：会构建一个递增的唯一值的新列表，并返回新列表values 中的值在旧列表中的个数 counts