Spróbuj 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])
Za pomocą można przyciąć tylko dolną połowę clip(0)
.
>>> a = numpy.array([1, 2, 3, -4, 5])
>>> a.clip(0)
array([1, 2, 3, 0, 5])
Możesz przyciąć tylko górną połowę za pomocą clip(max=n)
. (Jest to znacznie lepsze niż moja poprzednia sugestia, która obejmowała przekazanie NaN
do pierwszego parametru i użycie out
do wymuszenia typu.):
>>> a.clip(max=2)
array([ 1, 2, 2, -4, 2])
Innym ciekawym podejściem jest użycie where
:
>>> numpy.where(a <= 2, a, 2)
array([ 1, 2, 2, -4, 2])
Na koniec rozważ odpowiedź aix . Wolę clip
proste operacje, ponieważ dokumentuje się samoczynnie, ale jego odpowiedź jest preferowana w przypadku bardziej złożonych operacji.
numpy
? Próbowałaś klip metodę za
? Wbudowana funkcjanumpy.clip
daje mi ten sam błąd, ale metoda nie.Kolejne minimalistyczne rozwiązanie w Pythonie bez użycia numpy:
[0 if i < 0 else i for i in a]
Nie ma potrzeby definiowania żadnych dodatkowych funkcji.
a = [1, 2, 3, -4, -5.23, 6] [0 if i < 0 else i for i in a]
plony:
[1, 2, 3, 0, 0, 6]
źródło
for
zastosowanie go do każdego elementu listy, umieszczenie go po, oznacza, że tylko wtedy, gdy warunek jest spełniony, trafia do wynikowej listy.i < 0 ? 0 : i
w C) wewnątrz rozumienia list. Umieść nawiasy, aby było wyraźniej[(0 if i < 0 else i) for i in a]
. Wstawienie if after powoduje użycie części filtrującej konstrukcji wyrażenia listy.[(i) for i in a if i < 0]
zwróci tylko listę elementów, które są mniejsze od zera.I jeszcze jedna możliwość:
In [2]: a = array([1, 2, 3, -4, 5]) In [3]: where(a<0, 0, a) Out[3]: array([1, 2, 3, 0, 5])
źródło
Oto sposób na zrobienie tego w Pythonie bez NumPy. Utwórz funkcję, która zwraca to, czego chcesz i użyj wyrażenia listowego lub funkcji mapy .
>>> 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]
źródło