博客

查找NumPy数组中的唯一行

  • Post category:Python

查找NumPy数组中的唯一行可以通过以下步骤完成：

步骤 1：导入 NumPy 模块

使用 import 命令导入 NumPy 模块，以便能够使用 NumPy 数组。

import numpy as np

步骤 2：创建 NumPy 数组并去重

使用 np.unique() 函数对 NumPy 数组进行去重。这个函数可以返回一个数组的唯一值，并且可以根据需要返回这些值的索引或计数。

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

输出结果为：

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

在这个示例中，我们创建了一个二维数组[[1,2,3], [4,5,6], [1,2,3], [4,5,6]]。使用 np.unique() 函数按行进行去重，并将唯一的行存储在 unique_arr 数组中。

步骤 3：输出结果

输出唯一的行。

print(unique_arr)

完成以上步骤后，程序将输出唯一的行。以上面示例中的程序输出为例，将输出：

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

下面提供一个更加具体的示例：

my_arr = np.array([[1, 2, 3], [4, 5, 6], [1, 2, 3], [10, 11, 12], [4, 5, 6]])
unique_arr = np.unique(my_arr, axis=0)
print(unique_arr)

输出结果为：

[[ 1  2  3]
 [ 4  5  6]
 [10 11 12]]

在这个示例中，我们使用 np.unique() 函数对数组 [[1, 2, 3], [4, 5, 6], [1, 2, 3], [10, 11, 12], [4, 5, 6]] 进行去重，并将唯一的行存储在 unique_arr 数组中。最后，程序输出唯一的行。

以上是查找 NumPy 数组中的唯一行的完整攻略及示例，希望可以对你有所帮助。

Tags: