在python中比较单个元素和数组时如何选择max?
例如:i = 5&a = np.array([3,9,1,4,6])
预期结果是数组= [5,9,5,5,6]
通过比较i和a的每个元素并选择最大值。
除了用于循环,有更快的矢量化方式吗?
您正在寻找的是numpy.maximum:
numpy.maximum
import numpy as np i = 5 a = np.array([3,9,1,4,6]) print(np.maximum(i,a)) # [5,9,5,5,6]