如何使用索引或掩碼添加兩個numpy.array

Question

我有兩個數組，如下所示。

 a=numpy.array((1,2,3,4))
 b=numpy.array((1,2,3,4))

我只想添加前兩個元素。 我該如何使用蒙版[True, True, False, False]或索引[0,1]添加b=(2, 4,3,4)

Answer 1

很簡單：

b[:2] += a[:2]

要么：

mask = numpy.array((True, True, False, False))
b[mask] += a[mask]