我有两个数组,如下所示。
a=numpy.array((1,2,3,4))
b=numpy.array((1,2,3,4))
我只想添加前两个元素。如何使用蒙版[True, True, False, False]
或indice [0,1]
执行此操作
添加b=(2, 4,3,4)
答案 0 :(得分:1)
简单地说:
b[:2] += a[:2]
或者:
mask = numpy.array((True, True, False, False))
b[mask] += a[mask]