在Python3.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返回一个索引错误。怎么了?
发布于 2018-06-04 01:58:17
您可以直接使用numpy
添加向量
def vadd(a, b):
return a + b
为了安全起见,您可以进行isinstanceof
检查,以确保a,b
确实是numpy
数组。
https://stackoverflow.com/questions/50669320
复制相似问题