python怎么获取元素在array中的下标?

猿友 2021-07-29 13:45:10 浏览数 (7393)
反馈

在numpy中,我们可以使用array作为一个数组(或者称为列表)的容器。列表获取一个元素的下标相信各位小伙伴都清楚该如何操作,但是numpy的array怎么获取元素的下标有些小伙伴可能就不太了解了,接下来我们就来介绍python获取array的下标是怎么操作的吧。

用numpy创建一个array列表

>>>import numpy
>>>vector = numpy.array([5,10,15,20])
>>>print(vector)

 [5 10 15 20]

使用numpy的where方法寻找该数组下的某个值

>>>a = numpy.where(vector==10)
>>>print(a)

 (array([1], dtype=int64),)

返回数据是对应的下标

该下标被包裹在一个tuple和一个list中。我们可以用a[0][0]轻松取出来:

>>>print(a[0][0])

 1

补充:python如何获取列表中每个元素的下标位置

第一种方法

for i in range(len(numbers)):
print('({0}, {1})'.format(i, numbers[i]))

第二种方法

使用enumerate函数

numbers = [10, 29, 30, 41]
for index, value in enumerate(numbers):
print(index, value)

第三种

也是最有效的index内置函数

list1 = [1,22,31,4,6,7,8,23,5,89,90]
f1 = list1.index(22)
f2 = list1.index(5)
print(f1) #1
print(f2) #8

以上就是python获取array的下标的全部内容了,希望能给大家一个参考,也希望大家多多支持W3Cschool


0 人点赞