数组的去重

首先初始化一个数组

temp = np.array([[1, 2, 3, 4], [3, 4, 5, 6]])

再使用去重的api

np.unique(temp)

输出出来

array([1, 2, 3, 4, 5, 6])

set(temp.flatten())  # 先变成一维的才能用set

输出出来

{1, 2, 3, 4, 5, 6}