在python中比较单个元素和数组时选择max

Question

在python中比较单个元素和数组时如何选择最大值？

例如： i = 5＆a = np.array（[3,9,1,4,6]）

预期结果是一个数组= [5,9,5,5,6]

通过将i与a的每个元素进行比较并选择最大值。

除了用于循环之外，还有矢量化更快的方法吗？

Answer 1

您要寻找的是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]