如何为numpy.where()提供多个条件

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

import numpy as np

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

# 提供多个条件
condition1 = arr > 2
condition2 = arr < 5

# 使用逻辑与运算符组合条件
result = np.where(condition1 & condition2, arr, 0)

print(result)  # 输出: [0 0 3 4 0]

import numpy as np

# 创建一个示例数组
arr = np.array([10, 20, 30, 40, 50])

# 定义多个条件
condition1 = arr > 20
condition2 = arr < 40
condition3 = arr % 10 == 0

# 使用逻辑与运算符组合条件
result = np.where(condition1 & condition2 & condition3, arr * 2, arr)

print(result)  # 输出: [10 20 60 40 50]