如何使用索引或掩码添加两个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]