python - SciPy (Python) 中向量的 Romberg 积分

Question

好吧，我做错了什么？Scipy 的 romberg 积分器应该能够处理矢量函数，但我得到一个错误......

import numpy
from numpy import array
from scipy import integrate

f = lambda x:array([1, x, x**2])
result = integrate.romberg(f, 0, 1, vec_func=True)
print(result)

Answer 1

的文件romberg说：

vec_func : 布尔型，可选

func 是否将数组作为参数处理（即它是否是“向量”函数）。默认为假。

所以，参数 x 可以是一个向量。