numpy 배열에서 음수 값 바꾸기

배열의 모든 음수 값을 0으로 바꾸는 간단한 방법이 있습니까?

NumPy 배열을 사용하여 수행하는 방법에 대한 완전한 블록이 있습니다.

예

a = array([1, 2, 3, -4, 5])

돌아와야 해

[1, 2, 3, 0, 5]

a < 0 제공합니다 :

[False, False, False, True, False]

이것이 내가 붙어있는 곳입니다.이 배열을 사용하여 원래 배열을 수정하는 방법.

절반 정도입니다. 시험:

In [4]: a[a < 0] = 0

In [5]: a
Out[5]: array([1, 2, 3, 0, 5])

시도해보십시오 numpy.clip:

>>> import numpy
>>> a = numpy.arange(-10, 10)
>>> a
array([-10,  -9,  -8,  -7,  -6,  -5,  -4,  -3,  -2,  -1,   0,   1,   2,
         3,   4,   5,   6,   7,   8,   9])
>>> a.clip(0, 10)
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

를 사용하여 아래쪽 절반 만자를 수 있습니다 clip(0).

>>> a = numpy.array([1, 2, 3, -4, 5])
>>> a.clip(0)
array([1, 2, 3, 0, 5])

으로 상단 절반 만자를 수 있습니다 clip(max=n). (이는 NaN첫 번째 매개 변수 에 전달 out하고 유형을 강제하는 데 사용 하는 이전 제안보다 훨씬 낫습니다 .) :

>>> a.clip(max=2)
array([ 1,  2,  2, -4,  2])

또 다른 흥미로운 접근 방식은 다음을 사용하는 것입니다 where.

>>> numpy.where(a <= 2, a, 2)
array([ 1,  2,  2, -4,  2])

마지막으로 aix 의 대답을 고려하십시오 . clip자체 문서화이기 때문에 간단한 작업을 선호 하지만 더 복잡한 작업에는 그의 대답이 더 좋습니다.

numpy를 사용하지 않는 또 다른 미니멀 한 Python 솔루션 :

[0 if i < 0 else i for i in a]

추가 기능을 정의 할 필요가 없습니다.

a = [1, 2, 3, -4, -5.23, 6]
[0 if i < 0 else i for i in a]

수율 :

[1, 2, 3, 0, 0, 6]

그리고 또 다른 가능성 :

In [2]: a = array([1, 2, 3, -4, 5])

In [3]: where(a<0, 0, a)
Out[3]: array([1, 2, 3, 0, 5])

다음은 NumPy없이 Python에서 수행하는 방법입니다. 원하는 것을 반환하는 함수를 만들고 목록 이해 또는 지도 함수를 사용합니다.

>>> a = [1, 2, 3, -4, 5]

>>> def zero_if_negative(x):
...   if x < 0:
...     return 0
...   return x
...

>>> [zero_if_negative(x) for x in a]
[1, 2, 3, 0, 5]

>>> map(zero_if_negative, a)
[1, 2, 3, 0, 5]