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

问题描述 投票:0回答:1

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

例如:i = 5&a = np.array([3,9,1,4,6])

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

通过比较i和a的每个元素并选择最大值。

除了用于循环,有更快的矢量化方式吗?

python numpy vectorization numpy-ndarray
1个回答
2
投票

您正在寻找的是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]
© www.soinside.com 2019 - 2024. All rights reserved.