MichaelBradley/nbody
1
0
Fork 0
Code Issues Pull requests Activity Actions

Correct radius dimensions

Browse source
This commit is contained in:
Michael Bradley 2023-10-07 18:41:29 -04:00
parent 3ebf5b9baf
commit d54c95347f
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
data.py
View file

@ -11,11 +11,11 @@ def parse_csv(filename: str):
lines = file.read().strip().splitlines()
pos = np.zeros((len(lines), 2))
vel = np.zeros((len(lines), 2))
rad = np.zeros((len(lines),))
rad = np.zeros((len(lines), 1))
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