Question

所以我有一个看起来像这样的Ball类：

   class Ball(object):
    def __init__(self,n,x0,y0,dx,dy,r,c):
        self.xc = x0
        self.yc = y0
        self.dx = dx
        self.dy = dy
        self.radius = r
        self.area = math.pi*((r)**2)
        self.color = c
        self.name = n
    def position(self):
        return (self.xc,self.yc)
    def move(self):
        self.xc+=self.dx
        self.yc+=self.dy
    def collide(self,o,new_name):
        x = ((self.area*self.xc)+(o.area*o.xc))/(self.area+o.area)
        y = ((self.area*self.yc)+(o.area*o.yc))/(self.area+o.area)
        dx = ((self.area*self.dx)+(o.area*o.dx))/(self.area+o.area)
        dy = ((self.area*self.dy)+(o.area*o.dy))/(self.area+o.area)
        if self.area >= o.area:
            c = self.color
        else:
            c = o.color
        area = (self.area+o.area)
        r = math.sqrt(area/(math.pi))
        return Ball(new_name,x,y,dx,dy,r,c)
    def does_collide(self,o):
        if math.hypot((self.xc-o.xc),(self.yc-o.yc))<=(self.radius+o.radius):
            return True
        else:
            return False

我正在我的主代码中运行一个while循环来模拟各种球，每个球在每次迭代时都按其dx和dy值移动。为此，我有一个名为球的列表，用于保存我所有的球对象。它看起来像这样：

    balls=[ball1,ball2,ball3,ball4...] and so forth

我想要做的是使用我的球类中的do_collide函数来检查两个球是否碰撞，如果他们这样做，我想从列表中删除两个球，并在列表中插入一个新球，由碰撞功能。碰撞函数创建一个新球，其x，y，dx和dy值是两个球的加权平均值，其颜色是最大球的颜色。

所以对于我列表中的所有球，如何使用does_collide函数主动检查是否有两个球发生碰撞并将其从列表中删除？我还想将新球添加到列表中，即碰撞功能的结果。

我尝试过这样的事情：

         for ball1 in balls:
            for ball2 in balls:
                if ball1.name!=ball2.name:
                    if ball1.does_collide(ball2) == True:
                        ball = ball1.collide(ball2,(int(N)+1))
                        balls.append(ball)
                        balls.remove(ball1)
                        balls.remove(ball2)

但这似乎非常混乱，每次发生碰撞时都会崩溃。

Answer 1

你关心订购吗？如果没有，这是你可以做到的一种方式。

import itertools

newballs = []
removed = set()
for b1, b2 in itertools.combinations(balls, 2):
    if b1 not in removed and b2 not in removed:
        if b1.does_collide(b2):
            removed.add(b1)
            removed.add(b2)
            newballs.append(b1.collide(b2))

balls = [b for b in balls if b not in removed] + newballs

Answer 2

让你的外循环遍历整个列表，但让你的内循环只从当前的外循环球+ 1运行到列表的末尾。显然，如果球5（例如）与球8碰撞（例如），则球8与球5碰撞，因此您不必检查两者。

此外，您正在迭代列表的同时从列表中删除项目。这不安全。我不是在循环内部进行删除，而是在循环内设置标志，指示应该删除哪些球，然后在另一个循环中删除它们。

如何检查列表中的球是否发生冲突，并将其从列表中删除

2 个答案: