Вопрос об изменении значений одного массива на основе значений в другом массиве - PullRequest
0 голосов
/ 13 октября 2019

Рассмотрим двумерный массив NumPy, 1D массив массивов и константу:

arr1 = [[ 4  4]       arr2 = [ 1  7]        k = 2
        [ 3  6]
        [ 7 10] 
        [-2  6]
        [-1  6]
        [-8  8]]

Вот что мне нужно сделать: если абсолютное значение значений в arr1 [:, 0] находится вarr2, тогда мне нужно вычесть k из соответствующих значений в arr1 [:, 1]. Окончательный результат должен быть:

arr1 = [[ 4  4]      
        [ 3  6]
        [ 7  8] 
        [-2  6]
        [-1  4]
        [-8  8]]

Спасибо.

1 Ответ

0 голосов
/ 14 октября 2019

Все еще учусь, но, похоже, это работает:

print(arr1[np.in1d(abs[:,0]), arr2), 1] -= k)
...