如何使用索引或掩码添加两个numpy.array

时间:2014-03-26 09:29:05

标签: python numpy

我有两个数组,如下所示。

 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)

1 个答案:

答案 0 :(得分:1)

简单地说:

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

或者:

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