查找NumPy数组中的唯一行

  • Post category:Python

要查找NumPy数组中的唯一行,可以使用numpy.unique函数。该函数可以返回输入数组的唯一行。

以下是查找NumPy数组唯一行的完整攻略:

1. 导入NumPy模块

要使用NumPy模块的函数,首先需要导入该模块。

import numpy as np

2. 创建NumPy数组

在使用NumPy函数之前,需要创建输入数组。

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

3. 查找唯一行

使用numpy.unique函数找到唯一行。

unique_rows = np.unique(a, axis=0)

在这里,axis=0表示按行组合唯一值。如果axis=1,则按列组合唯一值。

4. 打印唯一行

打印唯一行数组。

print(unique_rows)

执行以上代码,输出:

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

示例1:使用一维数组

以下示例展示了在一维数组中查找唯一值。

a = np.array([1,2,3,4,5,3,2,4])
unique_rows = np.unique(a)
print(unique_rows)

在这里,唯一的行数组只包含输入数组中的不同值。

执行以上代码,输出:

[1 2 3 4 5]

示例2:使用二维数组

以下是在二维数组中查找唯一行的示例:

a = np.array([[1,2,3],[4,5,6],[1,2,3],[7,8,9]])
unique_rows = np.unique(a, axis=0)
print(unique_rows)

在这个例子里,唯一行数组是唯一的,因为输入数组中有重复行。

执行以上代码,输出:

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

这就完成了查找NumPy数组中唯一行的过程。