如何找出numpy array数组中的不同元素

例如在 array 数组[[[1,2,3,3,5], [5,9,18,100,2]]]中找出里面的不同元素
第一步就是把这个array数组用flatten铺平为([1,2,3,3,5, 5,9,18,100,2])
第二步就是用.tolist()进行转列表
第三步就是用set(list)进行不同元素的提取了

import numpy as np
a = np.array([[[1,2,3,3,5], [5,9,18,100,2]]])
print("a", a)
flatten_a = a.flatten()
print("flatten_a", flatten_a)
list_a = flatten_a.tolist()
print("list_a", list_a)
set_a = set(list_a)
print("set_a",set_a)

下面是上面代码块的各种print结果

a [[[  1   2   3   3   5]
    [  5   9  18 100   2]]]

flatten_a [  1   2   3   3   5   5   9  18 100   2]

list_a [1, 2, 3, 3, 5, 5, 9, 18, 100, 2]

set_a {1, 2, 3, 100, 5, 9, 18}

所以set_a里面就是数组a里面包含的不同元素了