在提到的代码之后,如何在其中插入函数作为参数?
我正在从Matlab迁移到Python,我想将一个函数以与函数参数相同的代码传递给另一个函数,但是我试图通过向下编写代码来做到这一点,但这给了我一个错误。我知道在Matlab中有一个函数运算符,可以通过@将函数插入另一个函数中。但是我该如何在python中做到呢?
import numpy as np
x = np.linspace(0,1,6)
def foo(y):
return y*np.sin(y)
def LoadVec1D(x, f):
n = len(x) - 1
b = np.zeros((n+1,1))
for i in range(n):
h = x[i+1] - x[i]
b[i] += f([x[i]])*h/2
b[i+1] += f([x[i+1]])*h/2
return b
print('b= ', LoadVec1D(x,foo(x)))