在 NumPy 数组中查找唯一行

在 NumPy 数组中查找唯一行是一个常见的需求。这可以通过使用 np.unique 函数实现，该函数可以返回数组中唯一的行。

代码示例

下面是一个示例数组：

import numpy as np

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

要查找唯一行，可以使用 np.unique 函数及其 axis 参数。使用 axis=0 将按行而非按列查找唯一值。

unique_rows = np.unique(arr, axis=0)
print(unique_rows)

输出结果如下所示：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

这样，我们就成功地查找了数组中唯一的行。

需要注意的是，如果想要保留原始数组的顺序，可以使用 np.unique 函数的 return_index 参数返回唯一行的索引，然后再使用这些索引从原始数组中选择对应的行。

结论

在 NumPy 数组中查找唯一行是一个非常简单的任务，只需要使用 np.unique 函数及其 axis 参数即可。要保留原始数组的顺序，可以使用 return_index 参数返回唯一行的索引，然后再使用这些索引从原始数组中选择对应的行。