Creating multiple class objects using a loop

Question

I have the following code:

N=200



class Nbody:

num_of_bodies = 0

def __init__(self, x, y, z, vx, vy, vz):
    self.x = x
    self.y = y
    self.z = z
    self.vx = vx
    self.vy = vy
    self.vz = vz

    Nbody.num_of_bodies +=1

def position(self):
    return '{}{}{} {}{} {}{}'.format('(',self.x,',', self.y,',', self.z,')')

nbody_1 = Nbody(random.random(), random.random(), random.random(), 0, 0, 0)
nbody_2 = Nbody(random.random(), random.random(), random.random(), 0, 0, 0)

print(nbody_1.position())
print(nbody_2.position())


print(Nbody.num_of_bodies)

I want to use a loop to create N number of n bodies instead of having to create them manually i.e where nbody_1 and nbody_2 are.

So for instance a loop that for N=200 creates 200 nbody_(i), where i is an integer step between and including 1 and 200.

For starters, you don't want 200 separate variables; use a list or a dict to store the objects instead. — chepner
– chepner, Commented Oct 11, 2018 at 15:22

Open AI - Opting Out · Accepted Answer · 2018-10-11 14:56:22Z

1

You can use a list comprehension:

nbodies = [Nbody(random.random(), random.random(), random.random(), 0, 0, 0)
           for i in range(N)]

answered Oct 11, 2018 at 14:56

Open AI - Opting Out

24.3k7 gold badges65 silver badges102 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Creating multiple class objects using a loop

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related