经纬恒润 标记重复元素 python

经纬恒润 标记重复元素 python

2022.09.15

题目描述

在给定的numpy数组中找到重复的条目(第二次出现以后)，并将它们标记为True(第一次出现应该为False)。

输入

[[0,0,3,0],[2,4,2,2]]
1

输出

[[false,true, false,true],[false,false,true,true]]
1

解题思路

numpy.unique的使用方法
numpy.unique 函数用于去除数组中的重复元素。
numpy.unique(arr, return_index, return_inverse, return_counts)

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

import numpy as np
def search(ori_array):
    res = []
    for i in range(len(ori_array)):# 列表的行数
        a = ori_array[i]# 列表中的每一行
        b=np.full(len(a),True) # b初始为True,表示第二次出现
        vals,counts=np.unique(a,return_index=True)# 函数用于去除数组中的重复元素
        # vals表示去重后的数组[0,3] [2,4]
        # counts表示去重后元素在原来所在的索引[0 2]--0在下标0位置，3在下标2位置；[0 1]--2在下标0位置，4在下标2位置
        b[counts] = False 
        res.append(list(b)) # 转换成列表添加到输出，不转类型的话b是array类型
    return res
ori_array=[[0,0,3,0],[2,4,2,2]]
print(search(ori_array))
1
2
3
4
5
6
7
8
9
10
11
12
13
14