在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]
来获取。
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,请发送邮件至 55@qq.com 举报,一经查实,本站将立刻删除。转转请注明出处:https://www.szhjjp.com/n/836685.html