将矢量化函数应用于数组numpy

Question

我想将一个函数应用于数组，但是当我调用它时，它会生成一个错误，因为该函数只进入数组的第一个组件

import numpy as np

def fObjetivo(x):
    print(x)
    return x[0] + x[1] + x[2]

fObjetivo = np.vectorize(fObjetivo)

a = np.array([1,2,3])

print(fObjetivo(a))

错误在哪里？

Answer 1

在这种情况下，数组a被矢量化并一次发送一个元素。声明为二维数组[：，3]，然后发送为[0]。它可能有用。