SciPy中的向量罗姆伯格积分（Python）

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 : 布尔值，可选

这个参数表示函数是否可以处理数组作为输入（也就是说，它是否是一个“向量”函数）。默认值是 False。

所以，这里的参数 x 可以是一个数组。