python输出数组的索引号_python数组查找某个值的位置

激活谷笔记 • 2025-05-17 10:42 • 阅读 120

在Python中，获取数组（列表）中素的索引值可以通过以下几种方法：

1. 使用`index（）`方法：

 arr = [1, 2, 3, 4, 5] index = arr.index（3） print（index） 输出：2

`index（）`方法返回指定素在列表中第一次出现的索引。

2. 使用列表推导式和`enumerate（）`函数：

 arr = [1, 2, 3, 4, 3, 5] target = 3 indices = [i for i, x in enumerate（arr） if x == target] print（indices） 输出：[2, 4]

当需要找到所有匹配素的索引时，可以使用列表推导式结合`enumerate（）`函数遍历列表，并筛选出所有匹配素的索引值。

3. 使用`numpy`库的`argwhere`函数：

 import numpy as np x = np.array（[[0, 1, 2], [3, 4, 5]]） indices = np.argwhere（x > 1） print（indices） 输出：array（[[0, 2], [1, 0], [1, 1], [1, 2]]）

`argwhere`函数可以找到数组中所有满足条件的素的索引。

4. 自定义函数`get_index`处理任意维度的数据集：

 import numpy as np def get_index（source, target）: if type（source） is not np.ndarray: source_arr = np.array（source） else: source_arr = source if type（target） is not np.ndarray: target_arr = np.array（target） else: target_arr = target source_dim = len（source_arr.shape） target_dim = len（target_arr.shape） 根据数据集维度进行处理 ...

自定义函数`get_index`可以处理任意维度的数据集，并返回指定素的索引。

请根据你的具体需求选择合适的方法来获取数组中素的索引值

python输出数组的索引号_python数组查找某个值的位置

相关推荐