python 返回数组内所有 大于1值的索引

在Python中，如果你有一个数组或列表，想要找出其中所有大于1的元素对应的索引，你可以使用`numpy`库的`where()`函数或者`list comprehension`结合`enumerate()`。以下是两种方法的示例：

**方法一：使用numpy.where()**

import numpy as np

arr = [0.5, 1.2, 2.3, 0.9, 4.5]
indices = np.where(arr > 1)[0]

print(indices)  # 输出：[1, 2, 4]

在这个例子中，`np.where(arr > 1)`返回一个包含True和False的布尔数组，非零索引对应的位置就是大于1的元素。

**方法二：使用list comprehension和enumerate()**

arr = [0.5, 1.2, 2.3, 0.9, 4.5]
indices = [i for i, x in enumerate(arr) if x > 1]

print(indices)  # 输出：[1, 2, 4]

这里`enumerate(arr)`会提供每个元素及其索引，筛选出满足条件的索引。

相关问题

python 数组中大于某个值的索引

可以通过enumerate()函数和列表推导式来实现。

示例代码如下：

my_list = [1, 2, 3, 4, 5, 6]
threshold = 3
result = [index for index, value in enumerate(my_list) if value > threshold]
print(result)  # 输出 [3, 4, 5]

其中，`my_list` 是需要查找的列表，`threshold` 是设定的阈值。使用enumerate()函数可以同时获取到列表中元素的索引和值。列表推导式中的 `if value > threshold` 条件语句用来筛选出大于阈值的元素索引。

布尔索引，返回数组中所有大于5的元素

假设要操作的数组为 `arr`，可以使用如下布尔索引操作来返回数组中所有大于5的元素：

arr[arr > 5]

这个操作会先生成一个布尔数组，其中 `arr > 5` 的位置为 `True`，其余位置为 `False`。然后将这个布尔数组作为索引传递给 `arr`，返回所有对应位置为 `True` 的元素。