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

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