将数组的元素与标量进行比较,并在Python中获得最大值

Question

我想将数组的元素与标量进行比较,并获得具有最大比较值的数组.那是我想打电话的

import numpy as np
np.max([1,2,3,4], 3)

并希望得到

array([3,3,3,4])

但我明白了

ValueError: 'axis' entry is out of bounds

我跑的时候

np.max([[1,2,3,4], 3])

我明白了

[1, 2, 3, 4]

这是列表中的两个元素之一,而不是我寻求的结果.有没有像其他内置函数一样快速的Numpy解决方案？

Answer 1

这已经内置到numpy功能np.maximum:

a = np.arange(1,5)
n = 3

np.maximum(a, n)
#array([3, 3, 3, 4])

这不会改变a:

a
#array([1, 2, 3, 4])

如果你想像@ jamylak的答案那样改变原始数组,你可以a输出:

np.maximum(a, n, a)
#array([3, 3, 3, 4])

a
#array([3, 3, 3, 4])

文件:

maximum(x1, x2[, out])

元素最大的数组元素.
相当于np.where(x1 > x2, x1, x2)但更快,并做适当的广播.