如何用NumPy获得数组的唯一元素

  • Post category:Python

获取数组中唯一元素可以利用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()函数成功获取数组中唯一的元素或者子数组。