有关基于另一个数组中的值修改一个数组的值的问题

时间:2019-10-13 00:20:11

标签: python-3.x numpy-ndarray

考虑一个2D numpy数组,一个1D numpy数组和一个常数:

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

这是我需要做的:如果arr1 [:,0]中的值的绝对值在arr2中,那么我需要从arr1 [:,1]中的相应值中减去k。最终输出应为:

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

谢谢。

1 个答案:

答案 0 :(得分:0)

仍在学习,但这似乎可行:

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