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

使用NumPy的unique函数可以获得数组的唯一元素。

首先需要导入NumPy库：

import numpy as np

接下来，我们可以使用numpy.unique()函数来获取一个数组的唯一元素。

函数原型

numpy.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None)

参数说明

• ar: 必须是数组。如果不是，那么将展开成一个一维数组。

• return_index: 如果为True，返回新列表元素在旧列表中的位置（下标），并以列表形式储

• return_inverse: 如果为True，返回旧列表元素在新列表中的位置（下标），并以列表形式储

• return_counts: 如果为True，返回新列表元素在旧列表中出现的次数，并以列表形式储

• axis: 待操作的轴或者坐标

示例说明

1. 获取一维数组的唯一元素

x = np.array([1, 2, 3, 1, 4, 5, 2, 5, 8, 9])
unique_x = np.unique(x)
print(unique_x)

输出结果为：

[1 2 3 4 5 8 9]

1. 获取二维数组的唯一元素

y = np.array([[1, 2, 3], [1, 3, 4], [2, 5, 6], [1, 2, 3]])
unique_y = np.unique(y, axis=0)
print(unique_y)

输出结果为：

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

这里的 axis=0 表示对二维数组的第一个维度（也就是行）去重。

通过上述示例，我们可以发现 np.unique 函数可以轻松地获取不同维度的数组的不重复元素。可以应用在众多场景，比如数据处理、数据可视化等等。