如何找到Numpy数组中的值的索引

您所在的位置：网站首页 › python返回指定元素索引 › 如何找到Numpy数组中的值的索引

如何找到Numpy数组中的值的索引

2023-10-28 01:24| 来源: 网络整理| 查看: 265

如何找到Numpy数组中的值的索引

在这篇文章中，我们将使用where()方法查找Numpy数组中的元素的索引。它是用来指定条件中指定的特定元素的索引。

语法：numpy.where(condition[, x, y])

获取一个给定值的索引位置

在这里，我们找到3的所有索引和3的第一次出现的索引。

# import numpy package import numpy as np # create an numpy array a = np.array([1, 2, 3, 4, 8, 6, 7, 3, 9, 10]) # display index value of 3 print("All index value of 3 is: ", np.where(a == 3)) print("First index value of 3 is: ",np.where(a==3)[0][0])

输出:

All index value of 3 is: (array([2, 7], dtype=int64),) First index value of 3 is: 2 打印几个值的第一个索引位置

在这里，我们要打印所有存在于数值阵列中的数值的索引号。

# import numpy package import numpy as np # create an numpy array a = np.array([1, 2, 3, 4, 8, 6, 2, 3, 9, 10]) values = np.array([2, 3, 10]) # index of first occurrence of each value sorter = np.argsort(a) print("index of first occurrence of each value: ", sorter[np.searchsorted(a, values, sorter=sorter)])

输出:

index of first occurrence of each value: [1 2 9] 根据多个条件获取元素的索引。

获取数值小于20且大于12的元素的索引

# Create a numpy array a = np.array([11, 12, 13, 14, 15, 16, 17, 15, 11, 12, 14, 15, 16, 17, 18, 19, 20]) # Get the index of elements with value less # than 20 and greater than 12 print("Index of elements with value less\ than 20 and greater than 12 are: \n", np.where((a > 12) & (a < 20)))

输出:

Index of elements with value less than 20 and greater than 12 are: (array([ 2, 3, 4, 5, 6, 7, 10, 11, 12, 13, 14, 15], dtype=int64),) 在的Python循环中获取元素的索引

创建一个NumPy数组，并对数组进行迭代，将数组中的元素与给定的数组进行比较。如果元素匹配则打印索引。

import numpy as np # create numpy array elements a = np.array([2, 3, 4, 5, 6, 45, 67, 34]) # display element index where value = 45 index_of_element = -1 for i in range(a.size): if a[i] == 45: index_of_element = i break if index_of_element != -1: print("element index : ", index_of_element) else: print("The element not present")

输出:

element index : 5

【本文地址】

如何找到Numpy数组中的值的索引

如何找到Numpy数组中的值的索引

今日新闻

推荐新闻