可以使用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。