Как выполнить арифметику условного массива в массиве numpy?

Я пытаюсь получить лучшее сцепление с массивами numpy, поэтому у меня есть примерный вопрос, чтобы спросить о них:

Скажем, у меня есть массив numpy, называемый a. Я хочу выполнить операцию над тем, которая увеличивает все значения внутри нее, которые меньше 0, и оставляет остальных в покое. например, если бы у меня было:

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

Я бы хотел вернуться:

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

Какой самый компактный синтаксис для этого?

Спасибо!

Ответы

Ответ 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])

Ответ 2

Чтобы мутировать его:

a[a<0] += 1

Чтобы оставить исходный массив отдельно:

a+[a<0]

Ответ 3

Просто чтобы добавить выше, в массиве numpy я хотел вычесть значение, основанное на значении ascii, чтобы получить значение от 0 до 35 для ascii 0-9 и AZ, и мне пришлось писать циклы for, но пост выше показал мне, как сделать это коротким. Так что я подумал о том, чтобы опубликовать это здесь, благодаря посту выше.

Код ниже можно сделать коротким

i = 0
for y in y_train:
    if y < 58:
        y_train[i] = y_train[i]-48
    else :
        y_train[i] = y_train[i] - 55
    i += 1
i = 0
for y in y_test:
    if y < 58:
        y_test[i] = y_test[i]-48
    else :
        y_test[i] = y_test[i] - 55
    i += 1
# The shortened code is below
y_train[y_train < 58] -= 48
y_train[y_train > 64] -= 55

y_test[y_test < 58] -= 48
y_test[y_test > 64] -= 55