python使用numpy库替换两个np数组中已知索引的值

使用场景

已知数组1：arr1= [1,2,3,4,5,6,7,8,9]
已知数组2：arr2= [0,0,0,0,0,0,1,1,1]
需要将数组2中值为0的数替换数组1中对应位置的数。
结果：result = [0,0,0,0,0,0,7,8,9]

实现代码

arr1[np.where(arr2 == 0)] = 0

np.where(arr2 == 0)返回arr2中值为0的索引

爱了爱了，numpy库和opencv库真是完美组合~python相比于c++真的香