Python中unique函数的使用

unique函数的介绍

Python中的unique函数可以从一个数组或列表中删除所有重复元素，仅保留唯一元素。这是我们在处理数据集合时经常需要使用的一个功能。这个函数有多种实现方式，最常用的是使用set()和numpy库。

使用set()实现unique函数

使用set()函数可以简单地实现unique函数，因为在set中元素是不可重复的，直接将列表转换为set，然后再转换回list即可实现去重。

def unique(lst):
    """
    通过set()函数去重
    """
    return list(set(lst))

使用numpy库实现unique函数

Numpy库中也提供了unique函数，不仅可以实现去重，还可以返回元素出现的次数和索引。

import numpy as np

def unique(lst):
    """
    使用numpy库中的unique函数去重
    """
    return np.unique(lst)

使用unique函数的实例

以下是用set()和numpy库实现的unique函数的使用示例：

lst = [1,3,2,2,1,5,6,5]

# 使用set()实现
print(unique(lst)) # [1, 2, 3, 5, 6]

# 使用numpy库实现
print(unique(lst)) # ([1, 2, 3, 5, 6], array([0, 2, 1, 5, 6]), array([2, 2, 1, 2, 1]))

通过使用unique函数，我们可以得到一个不含重复元素的列表，方便我们进行后续的数据处理。