minor changes

2021-11-05 16:18:15 +01:00
parent 6987baaf3a
commit d5f948f4a1
4 changed files with 1 additions and 35 deletions
--- a/main.py
+++ b/main.py
@@ -27,7 +27,7 @@ def main():
    v = np.array([v1, v2, v3])

    bodylist = []
-    for i in range(2): # For a system of 2 objects
+    for i in range(m.shape[0]):
        bodylist.append(Body(m[i], q[i], v[i]))
    dyn_syst = System(bodylist)
    dyn_syst.COMShift()