在numpy中,可以使用函数numpy.nonzero()
来创建下一个有效布尔值的索引数组。
numpy.nonzero()
函数返回输入数组中非零元素的索引。它接受一个数组作为参数,并返回一个包含两个数组的元组,分别表示非零元素的行索引和列索引。
下面是一个示例代码:
import numpy as np
arr = np.array([True, False, True, False, True])
index_array = np.nonzero(arr)
print(index_array)
输出结果为:
(array([0, 2, 4]),)
这表示在输入数组arr
中,索引为0、2、4的位置上的元素为True。
在numpy中,布尔值的索引数组可以用于选择数组中满足特定条件的元素。例如,可以使用布尔值的索引数组来选择所有为True的元素。
关于numpy的更多信息和使用方法,可以参考腾讯云的相关产品和文档:
领取专属 10元无门槛券
手把手带您无忧上云