Jak wykonać arytmetyczną tablicę warunkową w tablicy numpy?

Question

Próbuję uzyskać lepszą przyczepność na numpy tablice, więc mam próbkę pytanie, aby zapytać o nich:

Powiedz, że mam tablicę numpy o nazwie. Chcę wykonać operację, która inkrementuje wszystkie wartości w środku, które są mniejsze niż 0 i pozostawia resztę w spokoju. Na przykład, jeśli miałem:

a = np.array([1,2,3,-1,-2,-3]) 

chciałbym wrócić:

([1,2,3,0,-1,-2]) 

Co jest najbardziej zwarta składnię tego?

Dzięki!

Answer 1

In [45]: a = np.array([1,2,3,-1,-2,-3]) 

In [46]: a[a<0]+=1 

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

Answer 2

mutować go:

a[a<0] += 1 

Aby pozostawić oryginalną tablicę sam:

a+[a<0]