26 lines
496 B
Python
26 lines
496 B
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
time = 0
|
|
|
|
|
|
class Body:
|
|
|
|
def __init__(self, position, mass=1):
|
|
# Initial parameters
|
|
self.mass = mass
|
|
self.position = np.array(position)
|
|
|
|
|
|
def plot_t(positions):
|
|
fig, ax = plt.subplots()
|
|
ax.scatter(positions[:, 0], positions[:, 1])
|
|
plt.show()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
body1 = Body([1, 1])
|
|
body2 = Body([-1, -1])
|
|
positions = np.vstack([body1.position, body2.position])
|
|
plot_t(positions)
|