该博客介绍了numpy库中的unique函数，用于去除数组重复元素并排序。该函数不仅能简化数组，还支持返回元素首次出现的索引、元素在原数组中的逆序索引及计数。通过实例展示了基本用法，并详细解释了return_index、return_inverse和return_count等进阶参数的用途。
1 功能简介
该函数是去除数组中的重复数字，并排序之后输出。这个函数的功能可以用集合操作实现，但是集合操作并不能够让输出结果按一定顺序排列（集合进行了哈希操作）。
2 例子
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]
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
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
