nonzero(a)、where(a) 和 argwhere(a) 之间的差异。什么时候用哪个

中满

nonzero(a)、where(a) 和 argwhere(a) 之间的差异

在NumPy库中，有三个函数可以用于查找数组中满足特定条件的元素，并返回这些元素的索引或位置。这三个函数分别是nonzero(a)、where(a)和argwhere(a)。尽管它们的功能有些相似，但它们在使用时有一些差异。

nonzero(a)函数

nonzero(a)函数用于返回数组a中非零元素的索引。它返回一个包含两个数组的元组，分别表示非零元素的行索引和列索引。这个函数非常适合用于查找数组中非零元素的位置。

下面是一个例子，演示了如何使用nonzero(a)函数：

Python
import numpy as np
a = np.array([[1, 0, 3],
              [0, 5, 0],
              [7, 0, 9]])
result = np.nonzero(a)
print(result)

输出结果为：

(array([0, 0, 1, 2, 2]), array([0, 2, 1, 0, 2]))

这个结果表示在数组a中，非零元素的行索引和列索引分别是[0, 0, 1, 2, 2]和[0, 2, 1, 0, 2]。

where(a)函数

where(a)函数用于返回满足条件的元素的位置。它接受一个条件数组作为参数，返回一个包含满足条件的元素位置的元组。这个函数非常适合用于根据条件查找数组中的元素。

下面是一个例子，演示了如何使用where(a)函数：

Python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
result = np.where(a > 3)
print(result)

输出结果为：

(array([3, 4]),)

这个结果表示数组a中大于3的元素的位置是[3, 4]。

argwhere(a)函数

argwhere(a)函数与where(a)函数类似，不同之处在于它返回的是满足条件的元素的索引，而不是位置。它接受一个条件数组作为参数，返回一个包含满足条件的元素索引的二维数组。这个函数非常适合用于查找数组中元素的索引。

下面是一个例子，演示了如何使用argwhere(a)函数：

Python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
result = np.argwhere(a > 3)
print(result)

输出结果为：

[[3]
 [4]]

这个结果表示数组a中大于3的元素的索引是[[3], [4]]。

使用哪个函数？

根据以上的介绍和示例，我们可以出使用这三个函数的一般规则：

- 如果想要查找数组中非零元素的位置，应该使用nonzero(a)函数。

- 如果想要根据条件查找数组中元素的位置，应该使用where(a)函数。

- 如果想要查找数组中元素的索引，应该使用argwhere(a)函数。

根据具体的需求和问题，选择合适的函数可以更高效地进行数组元素的查找和处理。

通过本文的介绍，我们了解了nonzero(a)、where(a)和argwhere(a)函数之间的差异。这三个函数在NumPy库中用于查找数组中满足特定条件的元素，并返回它们的索引或位置。根据具体的需求和问题，选择合适的函数可以更高效地进行数组元素的查找和处理。