如何在类中使用“lambda”作为输入

如何解决如何在类中使用“lambda”作为输入

我想使用 lambda 作为名为 Integrator 的类函数的输入。在 Integrator 类内部，应根据当前状态调用对象。我不知道如何将这些对象引入 Integrator 类。 你能告诉我如何在另一个类中调用创建的对象吗？ 下面是主要的模拟程序：

# create LJ force object
lj_object = LennardJones(self.sigma,self.epsilon,self.compmethod,self.r_cut,self.Box_len)

# create spring force object
sp_object = InterMolecularForce (self.oh_len,self.k_b,self.tet_eq,self.k_tet)

# create Integrator object
integrator_object = Integrator (O_mass,H_mass)

for i in range (grid.shape[0]-1) :

        timespan = (grid [i],grid [i+1])                       

        lj_force =lj_object (new_postate)

        sp_force = sp_object(new_postate)

        new_postate[i+1],new_veLocity[i+1] = integrator_object (new_postate,new_veLocity,lambda new_postate: lj_object (new_postate) + sp_object(new_postate),timespan)

return new_postate,new_veLocity

积分器是：

# calculate half step momenta
momenta_half_step = diag_mass * veLocity + (force * (timespan[1] - timespan[0]) / 2)
position_full_step = posate + (timespan[1] - timespan[0]) * np.dot (inv(mass_matrix),momenta_half_step) 

# calculate forces
lj_force = lj_object (position_full_step)
spring_force = sp_object(position_full_step)
force = lj_force + spring_force

momenta_full_step = momenta_half_step + ( timespan[1] - timespan[0] ) * force / 2

解决方法

我有一种强烈的感觉，我错过了这里的重点。但是，无需尝试您的代码正在执行的任何复杂的应用数学，这里有一个简单的、准系统的方法来传递在 lambda 语句中定义的函数并将其传递给要使用的类。

class Operation:
    def __init__(self,operator):
        self.operator = operator
    def compute(self,*operands):
        return self.operator(operands[0],operands[1])


>>> addition=Operation(operator=lambda a,b: a+b)
>>> multiplication=Operation(operator=lambda a,b: a*b)
>>> addition.compute(3,5)
8
>>> multiplication.compute(3,5)
15

这是故意简化和设计的：它不进行类型检查，既不处理也不拒绝非二元运算符，总而言之，它不是很有用。它的唯一目的是展示如何调用传递给类的函数，这似乎是您问题的基本点。