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++真的香

©️2020 CSDN 皮肤主题: 黑客帝国 设计师:上身试试 返回首页