温馨提示×

numpy如何根据值找索引

numpy

小亿

142

2023-12-15 20:47:14

栏目: 编程语言

开发者测试专用服务器限时活动，0元免费领，库存有限，领完即止！点击查看>>

你可以使用numpy.where()函数来根据值找到索引。该函数返回一个包含满足条件的元素的索引的元组。

下面是一个示例：

import numpy as np

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

# 找到值为2的元素的索引
indexes = np.where(arr == 2)

print(indexes)

输出结果为：

(array([1, 3]),)

这表示索引1和索引3处的元素的值为2。请注意，np.where()返回的是一个元组，因此可以使用indexes[0]来获取索引的数组。

如果你想根据多个条件来查找索引，你可以将多个条件组合起来，例如：

indexes = np.where((arr == 2) & (arr < 4))

这将找到值为2且小于4的元素的索引。

0 赞

0 踩

最新问答