Python中如何找到列表或数组中True值的索引位置

当我们使用Python编写程序时，经常需要在列表或数组中找到某些条件的元素的索引位置。在这里，我们将讨论如何找到一个列表或数组中True值的索引位置。

列表中True值的索引位置

假设我们有一个包含一些布尔值的列表，我们想要找到所有True值的索引位置。在Python中，可以使用“enumerate”和“list comprehension”来完成此操作。以下是代码示例:

my_list = [True, False, True, False, True, True, False]
result = [index for index, value in enumerate(my_list) if value]
print(result)

输出结果：

[0, 2, 4, 5]

• 在这里，我们使用"enumerate"函数获取元素的索引和值。然后使用List Comprehension筛选True元素的索引。

数组中True值的索引位置

对于NumPy数组，可以使用“numpy.where”函数来找到所有True值的索引。以下是代码示例：

import numpy as np
my_array = np.array([True, False, True, False, True, True, False])
result = np.where(my_array)[0]
print(result)

输出结果：

[0, 2, 4, 5]

• 在这里，我们使用“np.where”返回满足条件的元素的索引数组，我们使用索引0获取索引数组。

综上所述，我们可以使用Python和NumPy引用程序来找到一个列表或数组中True值的索引位置。