如何将数组传递给Python中的函数?

时间:2018-06-03 17:51:28

标签: python numpy-ndarray

在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返回一个索引错误。怎么了?

1 个答案:

答案 0 :(得分:0)

您只需使用numpy直接添加向量:

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

为安全起见,您可以进行isinstanceof检查以确保a,b确实是numpy数组。