获取数组中唯一元素可以利用NumPy中的unique()函数实现。下面是获取数组唯一元素的完整攻略:
步骤一:导入NumPy模块
import numpy as np
步骤二:创建数组
arr = np.array([1, 2, 2, 3, 4, 4, 5, 5, 5])
上面是一个简单的例子,包含了重复元素。接下来将使用unique()函数来获得数组中的唯一元素。
步骤三:使用unique()函数并打印结果
unique_arr = np.unique(arr)
print(unique_arr)
输出结果:
[1 2 3 4 5]
unique()函数返回一个已排序的包含数组唯一值的新数组。
下面再给出一个更为复杂的例子:
arr = np.array([[1, 2, 3], [4, 4, 5], [6, 6, 7]])
上面数组包含了多个重复的子数组。我们可以使用unique()函数来获取数组中唯一的子数组:
unique_arr = np.unique(arr, axis=0)
print(unique_arr)
输出结果:
[[1 2 3]
[4 4 5]
[6 6 7]]
在这个例子中,由于设置了axis参数为0,unique()函数会返回一个包含唯一子数组的新数组。
综上所述,通过以上步骤,我们可以使用NumPy的unique()函数成功获取数组中唯一的元素或者子数组。