Сравнение элементов массива со скаляром и получение max в Python
Я хочу сравнить элементы массива со скаляром и получить массив с максимумом сравниваемых значений. Вот я и хочу позвонить
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 для того, что быстро, как и другие встроенные функции?
2 ответов
это уже встроено в 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])
Docs:
maximum(x1, x2[, out])Element-wise максимум элементов массива.
Эквивалентноnp.where(x1 > x2, x1, x2)но быстрее и не правильное радиовещание.
>>> import numpy as np
>>> a = np.array([1,2,3,4])
>>> n = 3
>>> a[a<n] = n
>>> a
array([3, 3, 3, 4])