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