可以使用numpy.where()
函數(shù)來查找數(shù)列中某個數(shù)的索引。
下面是一個例子,演示了如何找到數(shù)列中值為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]),)
可以看到,數(shù)字5在數(shù)列中的索引為4和7。請注意,np.where()
函數(shù)返回的是一個元組,其中第一個元素是包含索引的數(shù)組。
如果想要查找多個值的索引,可以使用numpy.in1d()
函數(shù)。下面是一個例子,演示如何找到數(shù)列中值為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]),)
可以看到,數(shù)字5和6在數(shù)列中的索引為4、5和7。