我想将数组的元素与标量进行比较,并获得具有最大比较值的数组。那是我想打电话的
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解决方案?
答案 0 :(得分:15)
这已经使用函数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])
Docs:
maximum(x1, x2[, out])
元素最大数组元素。
相当于np.where(x1 > x2, x1, x2)
,但更快,并进行适当的广播。
答案 1 :(得分:2)
>>> import numpy as np
>>> a = np.array([1,2,3,4])
>>> n = 3
>>> a[a<n] = n
>>> a
array([3, 3, 3, 4])