如何在numpy中创建一个以索引为条件的掩码？

import numpy as np

# 创建一个数组
arr = np.array([1, 2, 3, 4, 5])

# 定义条件：选择大于3的元素
condition = arr > 3

# 创建掩码
mask = condition

# 使用掩码选择元素
selected_elements = arr[mask]

print("原始数组:", arr)
print("掩码:", mask)
print("满足条件的元素:", selected_elements)

原始数组: [1 2 3 4 5]
掩码: [False False False  True  True]
满足条件的元素: [4 5]

# 组合多个条件
complex_condition = (arr > 2) & (arr < 5)
complex_mask = complex_condition
complex_selected_elements = arr[complex_mask]

print("复杂条件:", complex_condition)
print("复杂掩码:", complex_mask)
print("满足复杂条件的元素:", complex_selected_elements)

复杂条件: [False False  True  True False]
复杂掩码: [False False  True  True False]
满足复杂条件的元素: [3 4]