Numpy 如何通过行查找行索引

在使用Numpy处理数据时，我们经常需要根据行查找其在数组中的索引。本文将介绍如何使用Numpy来实现这个过程。

阅读更多：Numpy 教程

使用enumerate方法

常规的方法是使用Python的enumerate函数，这将返回一个列表，其中每个元素都是元组，包含列表中该元素的索引和该元素的值。我们可以使用此方法来查找特定行的索引：

import numpy as np

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

for index, row in enumerate(arr):
    if (row == [4, 5, 6]).all():
        print(index)

在上面的示例中，我们定义了一个数组，然后使用enumerate函数迭代该数组。一旦找到匹配的行，我们就打印所在的索引值。

使用where方法

另一种方式是使用Numpy的where函数。该函数返回几个数组，其中一个数组包含位置信息。

import numpy as np

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

index = np.where((arr == [4, 5, 6]).all(axis=1))[0][0]

print(index)

在上面的示例中，我们定义了一个数组，然后使用where函数查找匹配的行。在这种情况下，where函数返回的数组包含一行中所有布尔相等值的数组。我们使用all（axis=1）函数来检查每一行是否都相等。由于where函数返回一个数组，我们需要从中获取第一个元素（即要查找的行的位置）。

总结

通过以上介绍可以看出，使用enumerate方法和where方法都可以在Numpy中查找行索引。其中，使用where方法具有更高的效率，因为它仅遍历目标数组一次。不过在numpy中，对于一些小数据量的数组或者运行次数较少的索引操作两种方法都可行。