numpy

numpy怎么查找元素索引

小亿
235
2024-01-25 12:23:20
栏目: 编程语言
开发者专用服务器限时活动,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]来获取。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

相关推荐:numpy怎么获取元素索引

0
看了该问题的人还看了