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。