numpy

numpy如何根据值找索引

小亿
138
2023-12-15 20:47:14
栏目: 编程语言

你可以使用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
看了该问题的人还看了