Remove unneeded split

data.py

@@ -15,7 +15,7 @@ def parse_csv(filename: str):
     for i, [x, y, vx, vy, r] in enumerate(map(lambda l: map(float, l.split(',')), lines)):
         pos[i] = [x, y]
         vel[i] = [vx, vy]
-        rad[i] = [r]
+        rad[i] = r
     return pos, vel, rad
 
 

physics.py

@@ -7,9 +7,7 @@ G = 6.674e-11
 def rotations(a: np.ndarray):
     a2 = np.concatenate((a, a))
     for i in range(1, len(a)):
-        yield np.split(a2, [i, i + len(a)])[1]
-        # TODO: Compare performance
-        # yield np.roll(a, i)
+        yield a2[i: i + len(a)]
 
 
 def n_body(pos: np.ndarray, vel: np.ndarray, mass: np.ndarray):