From decedf5731cf5c45a86f047406a9a0c024981aaa Mon Sep 17 00:00:00 2001 From: Michael Bradley Date: Sat, 7 Oct 2023 18:57:30 -0400 Subject: [PATCH] Remove unneeded split --- data.py | 2 +- physics.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/data.py b/data.py index d9a2417..478c542 100644 --- a/data.py +++ b/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 diff --git a/physics.py b/physics.py index 0d73b9e..6b3fb0c 100644 --- a/physics.py +++ b/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):