如何在Python中將數組傳遞給函數？

Question

在Python 3.6中，我試圖構建將向量作為輸入和輸出向量的函數。 作為一個簡單的示例，請考慮以下代碼：

import numpy as np

def vadd(a,b):

    res = np.array([[0,0,0]])

    res[0] = a[0]+b[0]

    res[1] = a[1]+b[1]

    res[2] = a[2]+b[2]

    return res

當我對具有三個元素的np.array類型的對象運行此命令時，Python返回索引錯誤。 怎么了？

Answer 1

您可以直接使用numpy添加向量：

def vadd(a, b):
    return a + b

為了安全起見，您可以進行isinstanceof檢查，以確保a,b確實是numpy數組。