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

Add slightly better n_body_matrix

Browse source
This commit is contained in:
Michael Bradley 2023-10-07 21:10:37 -04:00
parent aca44fc3b0
commit e1b55dbad2
2 changed files with 22 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
data.py
View file

@ -3,7 +3,7 @@ import matplotlib.cm as cm
import matplotlib.pyplot as plt
import numpy as np
from physics import n_body
from physics import n_body_matrix
def parse_csv(filename: str):
@ -54,7 +54,7 @@ class Animator:
return self.scat,
def update(self, *_args, **_kwargs):
n_body(self.pos, self.vel, self.mass)
n_body_matrix(self.pos, self.vel, self.mass)
self.scat.set_offsets(self.pos)
return self.scat,