如何交换功能来自功能的元素？

Question

因此，我定义了一个称为“颗粒”的类：

class Particle:
    def __init__(self, posInit, momInit, spin):
        self.posInit = posInit
        self.momInit = momInit
        self.spin = spin
    def momfT(self, t):
        return self.momInit*(math.cos(t))-self.posInit*(math.sin(t))
    def posfT(self, t):
        return self.posInit*(math.cos(t))+self.momInit*(math.sin(t))

P = [Particle(posInit = i, momInit = j, spin = choice(["up", "down"])) for i,j in zip(Zinitial,Pinitial)]

我现在想做的是如果满足某个条件，请切换粒子的位置。 因此，类似以下内容：

if cond==True:
  P[1].posfT[t], P[2].posfT[t1] = P[2].posfT[t1], P[1].posfT[t]

但是以上内容不起作用，因为我试图将函数分配给一个值。

所以我不确定该怎么做？

Answer 1

看来您打算将粒子的位置和动力随时间存储在posft和momft （编辑：我以前以前以前都认为您只想要当前位置）。如果是这样，它们不应该是方法，而是属性。您还应该有单独的方法来修改这些值，随着T的发展。我建议这样修改您的班级：

class Particle:
    
    def __init__(self, posInit, momInit, spin):
        self.posInit = posInit
        self.momInit = momInit
        self.spin = spin
        self.momfT = dict()
        self.posfT = dict()
        self.calc_at_time(0)
    
    def calc_at_time(self, t):
        self.momfT[t] = self.momInit*(math.cos(t))-self.posInit*(math.sin(t))
        self.posfT[t] = self.posInit*(math.cos(t))+self.momInit*(math.sin(t))

注意：我假设POSFT和MOMFT使用t = 0初始化。我绝对缺乏知道这是否正确的知识，请根据需要检查并纠正。我在这里专注于代码。

然后，您可以通过调用calc_at_time（t）来在时间t上设置新位置

t = 1
t1 = 2
P[1].calc_at_time(t)
P[1].calc_at_time(t1)
P[2].calc_at_time(t)
P[2].calc_at_time(t1)

，并使用p [x] .posft [t]在时间t上访问P [X]的位置。

因此，您要做的事情现在应该有效：

if cond == True:
  P[1].posfT[t], P[2].posfT[t1] = P[2].posfT[t1], P[1].posfT[t]

含义：p [1]的p [t]的posft变成t1 p [2]的p [t]，并相互互惠。