python - 将数组的元素与标量进行比较并在 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)但速度更快，并且可以进行适当的广播。

Answer 2

>>> import numpy as np
>>> a = np.array([1,2,3,4])
>>> n = 3
>>> a[a<n] = n
>>> a
array([3, 3, 3, 4])