阅读量:0
在NumPy中，可以使用np.where()函数来查找元素索引。np.where()函数返回满足指定条件的元素的索引。以下是使用np.where()函数查找元素索引的示例：
import numpy as np arr = np.array([1, 2, 3, 4, 5, 6]) # 查找元素3的索引 index = np.where(arr == 3) print(index) # 输出 (array([2]),) # 查找元素大于4的索引 index = np.where(arr > 4) print(index) # 输出 (array([4, 5]),) # 查找元素不等于2的索引 index = np.where(arr != 2) print(index) # 输出 (array([0, 2, 3, 4, 5]),) 注意，np.where()函数返回的是一个元组，其中包含满足条件的元素的索引数组。如果要获取索引数组，可以使用index[0]来获取。
