python 如何获取元素在array中的下标

获取元素在array中的下标，通常可以通过Python内置模块numpy和list自带的方法来实现。

一、使用numpy模块

1. numpy.where()方法

numpy.where(condition, [x, y])

该方法返回满足条件的元素下标。

示例：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
index = np.where(arr == 3)
print(index) # 输出 (array([2]),)

1. numpy.argwhere()方法

numpy.argwhere(a)

该方法返回非零元素的下标。

示例：

import numpy as np

arr = np.array([0, 1, 0, 1, 0])
index = np.argwhere(arr == 1)
print(index) # 输出 [[1], [3]]

二、使用list自带的方法

1. list.index()方法

list.index(x[, start[, end]])

该方法返回元素在列表中的下标。

示例：

arr = [1, 2, 3, 4, 5]
index = arr.index(3)
print(index) # 输出 2

1. enumerate()函数

enumerate(iterable, start=0)

该函数返回一个枚举对象，同时返回下标和对应的元素。

示例：

arr = [1, 2, 3, 4, 5]
for index, value in enumerate(arr):
    if value == 3:
        print(index) # 输出 2

以上就是获取元素在array中的下标的两种方法，根据不同情况可灵活选择使用。