首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >如何在Python中将数组传递给函数?

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

Stack Overflow用户
提问于 2018-06-04 01:51:28
回答 1查看 51关注 0票数 0

在Python3.6中,我试图构建以向量作为输入和输出向量的函数。作为一个简单的示例,请考虑以下代码:

代码语言:javascript
复制
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返回一个索引错误。怎么了?

EN

回答 1

Stack Overflow用户

发布于 2018-06-04 01:58:17

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

代码语言:javascript
复制
def vadd(a, b):
    return a + b

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

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/50669320

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档