python numpy-執行向量和矩陣加法的函數

Question

我正在研究問我的問題：

如果可能，將兩個NumPy向量或矩陣加在一起。 如果不可能將兩個向量/矩陣加在一起（因為它們的大小不同），則返回False。

這是我的方法：

import numpy as np

def mat_addition(A, B):
    if A.shape != B.shape:
        return False
    else:
        return np.sum(A,B)

但是當我運行代碼進行測試時，它說

TypeError: only integer scalar arrays can be converted to a scalar index

有人可以告訴我我的代碼有什么問題嗎？

Answer 1

實際上， np.sum可以按您想要的方式使用。 您只需要將傳遞給np.sum的參數包裝在一個列表中：

import numpy as np

def mat_addition(A, B):
    if A.shape != B.shape:
        return False
    else:
        return np.sum([A,B])

a = np.arange(5*3).reshape(5,3)
b = np.arange(5*3, 5*3*2).reshape(5,3)
print(mat_addition(a,b))

輸出：

435

根據numpy.sum文檔，此函數將單個“ array_like”對象作為其第一個參數。 數組列表是一個完全有效的“ array_like”對象，因此上述代碼可以正常工作。