溫馨提示×

numpy怎么求數(shù)列中某個數(shù)的索引

小億
105
2023-11-24 04:07:15
欄目: 編程語言

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

0