如何将数组传递给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数组。