考虑一个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]]
谢谢。
答案 0 :(得分:0)
仍在学习,但这似乎可行:
print(arr1[np.in1d(abs[:,0]), arr2), 1] -= k)