numpy where返回一个数组。我只需要索引

numpy where函数是一个用于根据指定条件返回数组中满足条件的元素索引的函数。它可以根据条件返回一个新的数组，其中包含满足条件的元素的索引。

使用numpy where函数，可以按照以下方式获取满足条件的元素索引：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
condition = arr > 2

indices = np.where(condition)
print(indices)

输出结果为：

(array([2, 3, 4]),)

这表示满足条件的元素索引为2、3和4。

numpy where函数的参数和返回值如下：

numpy.where(condition[, x, y])

• condition：一个条件表达式，用于指定满足条件的元素。
• x, y：可选参数，用于指定满足条件和不满足条件的元素的替代值。如果不提供这两个参数，则返回满足条件的元素索引。

numpy where函数的应用场景包括但不限于：

• 数据过滤：根据条件过滤数组中的元素。
• 数组索引：获取满足条件的元素的索引，进而进行进一步的操作。
• 数组修改：根据条件修改数组中的元素值。

腾讯云提供了一系列与numpy相关的产品和服务，例如云服务器、云数据库、云存储等，可以满足各种云计算需求。具体产品和服务的介绍和链接地址可以在腾讯云官方网站上找到。