在NumPy中按位置将数组元素与另两个数组比较的向量化方法

你可以直接使用NumPy内置的np.minimum函数完成这个需求，这是完全向量化的实现，无需编写for循环，效率远高于Python原生循环。

代码示例 #

import numpy as np

# 定义数组（需转换为NumPy数组）
a_array = np.array([5031, 5038, 5040])
b_array = np.array([5050, 5055, 5870])
c_array = np.array([5611, 5743, 5780])

# 向量化计算每个位置b、c元素的较小值
result = np.minimum(b_array, c_array)
print(result)  # 输出: [5050 5055 5780]

说明 #

np.minimum会对输入的两个数组执行逐元素比较，直接返回每个索引位置上的较小值，底层基于C实现，处理大规模数组时性能优势明显。

内容的提问来源于stack exchange，提问作者Andy