2014-03-26 26 views
0

我有兩個數組,如下所示。如何添加兩個numpy.array索引或掩碼

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)

回答

1

它只是:

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

或者:

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