Python之元素去重

1. 一维数组去重

给定一个一维数组，如下：

a = [0, 1, 2, 3, 4, 3, 2, 3, 4, 8, 7, 5, 9, 0, 1, 6]

代码：

# -*- coding:utf-8 -*-
import numpy as np

if __name__ == '__main__':
    # 一维数组去重
    a = [0, 1, 2, 3, 4, 3, 2, 3, 4, 8, 7, 5, 9, 0, 1, 6]
    a = np.unique(a)
    print(a)

去重后的结果，如下：

[0  1  2  3  4  5  6  7  8  9]

2. 二维数组去重

给定一个二维数组，如下：

b = [[1, 2], [3, 4], [5, 6], [3, 4], [1, 2], [7, 8], [6, 6], [1, 2]]

代码：

# -*- coding:utf-8 -*-
import numpy as np

if __name__ == '__main__':
    # 二维数组去重
    b = [[1, 2], [3, 4], [5, 6], [3, 4], [1, 2], [7, 8], [6, 6], [1, 2]]
    s = set()

    for t in b:
        s.add(tuple(t))

    print(np.array(list(s)))

去重后的结果，如下：

[[1 2]
 [7 8]
 [6 6]
 [5 6]
 [3 4]]

可以将代码略微简化，如下：

# -*- coding:utf-8 -*-
import numpy as np

if __name__ == '__main__':
    # 二维数组去重
    b = [[1, 2], [3, 4], [5, 6], [3, 4], [1, 2], [7, 8], [6, 6], [1, 2]]
    print(np.array(list(set([tuple(t) for t in b]))))

可以发现结果一样：

[[1 2]
 [7 8]
 [6 6]
 [5 6]
 [3 4]]