获取第一个真值 numpy 的索引 - Python

在Python中，numpy是一种强大的科学计算库，在数据处理和分析中应用广泛。有时我们需要获取ndarray数组中第一个为真的元素的索引，可以使用numpy库提供的函数来实现。

实现方法

假设我们有一个ndarray数组arr，我们想要获取第一个为真的元素的索引，可以使用numpy库提供的nonzero函数。该函数返回一个元组，包含所有非零元素的索引。

import numpy as np

arr = np.array([0, 1, 2, 0, 3, 4, 5])
indexes = np.nonzero(arr)[0]

这样，我们就可以获取到第一个为真的元素的索引。如果要获取所有为真的元素的索引，可以使用以下代码：

import numpy as np

arr = np.array([0, 1, 2, 0, 3, 4, 5])
indexes = np.nonzero(arr)[0]
print(indexes)

输出结果为：

[1, 2, 4, 5, 6]

如果我们只需要获取第一个为真的元素的索引，可以使用以下代码：

import numpy as np

arr = np.array([0, 1, 2, 0, 3, 4, 5])
index = np.nonzero(arr)[0][0]
print(index)

输出结果为：

1

结论

通过使用numpy库提供的nonzero函数可以很方便地获取ndarray数组中第一个为真的元素的索引。在实际应用中，我们可以根据需要获取所有为真的元素的索引或只获取第一个为真的元素的索引。