Python去重操作及相关函数

栗子1：利用 set

L1 = (1, 1, 2, 2, 3, 3)
print(set(L1))

结果为： {1, 2, 3}

栗子2：利用 np.unique

import numpy as np
L1 = (1, 1, 2, 2, 3, 3)
print(np.unique(L1))

结果为： [1 2 3]

栗子3：利用循环

import numpy as np
L1 = (1, 1, 2, 2, 3, 3)
L2 = list(L1)
L3 = []

for i in L2:
    if i not in L3:
        L3.append(i)
print(L3)

结果为： [1, 2, 3]

以下来自博客：Python 二维数组元素去重 np.unique()函数的使用

### 4.2 元素去重
import numpy as np
# 4.2.1 一维数组去重
a = np.array([1, 2, 3, 4, 5, 5, 7, 3, 2, 2, 8, 8])
print('去重前：', a)
b = np.unique(a)
print('去重后：', b)
# 4.2.2 二维数组去重
c = np.array(((1, 2), (3, 4), (5, 6), (1, 3), (3, 4), (7, 6)))
print('去重前：', c)
# 方法一：利用set,set是无序不重复集合(推荐)
s = set() #创建空集合
for t in c:
    s.add(tuple(t)) #将数组转为元祖tuple，保证不被修改；再把元祖加入到集合中，完成去重
g = np.array(list(s)) # 将集合转换为列表，最后转为二维数组
print('去重后：', g)
# 合起来写
print('方法一去重后：', np.array(list(set(tuple(t) for t in c))))
# 方法二：转换为虚数
r, i = np.split(c, (1, ), axis=1)
print(r)
print(i)
x = r + i * 1j
# x = c[:, 0] + c[:, 1] * 1j
print('转换成虚数：', x)  # 变成一维数组，可使用np.unique()
print('虚数去重后：', np.unique(x))
print(np.unique(x, return_index=True, return_counts=True))
# return_index=True表示返回另一个参数，这个参数也是一个数组，
# array([0, 3, 1, 2, 5]表示去重后的每个元素在去重前的数组中的位置/下标
# return_counts=True表示返回各元素出现的频次
idx = np.unique(x, return_index=True)[1]
# 第[1]列的数组正是return_index=True返回的内容：在c中的位置/下标
print('二维数组去重：\n', c[idx]) #提取这些下标所在的元素

来源：CSDN

作者：北木.

链接：https://blog.csdn.net/weixin_43283397/article/details/104612852

标签