將陣列的值添加到numpy矩陣的行中

Question

我想將numpy陣列中的值添加到numpy矩陣的特定行中的值。

考慮：

A = [[0, 0], [0, 0]] 

b = [1, 1] 

我想在A. B加入A值的第一行中的預期結果是：

[[1, 1], [0, 0]] 

我使用 「+」 操作符嘗試過，但出現錯誤：

>>> import numpy 
>>> a = numpy.zeros(shape=(2,2)) 
>>> a 
array([[ 0., 0.], 
     [ 0., 0.]]) 
>>> b = numpy.ones(shape=(1,2)) 
>>> b 
array([[ 1., 1.]]) 

>>> a[0, :] += b 

Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
ValueError: non-broadcastable output operand with shape (2,) doesn't match the broadcast shape (1,2) 

這樣做的最佳方法是什麼？

Answer 1

b = [1, 1]和b = [[1, 1]]之間的差異。 a[0, :] += b失敗，因爲在這種情況下無法播放。

b如果可以包含可變數量行的話，可以使用b的長度取的a切片和添加b到它。

>>> a = numpy.zeros(shape=(2,2)) 
>>> b = numpy.ones(shape=(1,2)) 
>>> a[:len(b)] += b 
>>> a 
array([[ 1., 1.], 
     [ 0., 0.]]) 

或者，如果b包含那麼只有一行：

>>> a = numpy.zeros(shape=(2,2)) 
>>> a[0] += b[0] 
>>> a 
array([[ 1., 1.], 
     [ 0., 0.]]) 

Answer 2

a = np.zeros((2 , 2)) 
b = np.ones((1 ,2)) 
np.concatenate([b , np.array([a[0]])])