温馨提示×

numpy怎么求数列中某个数的索引

小亿
105
2023-11-24 04:07:15
栏目: 编程语言

可以使用numpy.where()函数来查找数列中某个数的索引。

下面是一个例子,演示了如何找到数列中值为5的元素的索引:

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6, 7, 5, 8, 9])
indices = np.where(arr == 5)

print(indices)

输出结果为:

(array([4, 7]),)

可以看到,数字5在数列中的索引为4和7。请注意,np.where()函数返回的是一个元组,其中第一个元素是包含索引的数组。

如果想要查找多个值的索引,可以使用numpy.in1d()函数。下面是一个例子,演示如何找到数列中值为5和6的元素的索引:

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6, 7, 5, 8, 9])
values = np.array([5, 6])
indices = np.where(np.in1d(arr, values))

print(indices)

输出结果为:

(array([4, 5, 7]),)

可以看到,数字5和6在数列中的索引为4、5和7。

0