博客

要查找NumPy数组中的唯一行，可以使用numpy.unique函数。该函数能够返回一个具有唯一行的新数组，并且可以指定在哪些列上进行去重行的判断。

具体操作如下：

1. 导入numpy模块

import numpy as np

1. 创建一个二维数组

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

1. 调用numpy.unique函数，并指定在所有列上去重

unique_arr = np.unique(arr, axis=0)

这里的axis参数表示在哪个维度上进行去重。axis=0表示在列上去重，axis=1表示在行上去重。

1. 打印输出结果

print(unique_arr)

运行以上代码，输出结果如下：

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

可以看到，该二维数组的唯一行为[1, 2]、[3, 4]、[5, 6]。

下面给出一个更具有挑战性的示例：

arr = np.array([[1, 2, 3], [1, 2, 4], [1, 2, 3], [1, 2, 5], [2, 3, 4]])
unique_arr = np.unique(arr[:, :2], axis=0)
result = [arr[i] for i in range(len(arr)) if np.array_equal(unique_arr, arr[i, :2])]
print(result)

这里的示例中，我们要求的是在前两列不重复的情况下，获取唯一行。运行结果为[[1, 2, 3], [1, 2, 4], [2, 3, 4]]，即第三列不同的行组成的新数组。在这个示例中，我们使用了切片和循环语句，可以看到，除了numpy.unique函数之外，我们还使用了一些其他的NumPy操作。