Как передать массив функции в 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 массивы.