20 lines
383 B
Python
20 lines
383 B
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
time = 0
|
|
# Initial conditions
|
|
mass_b1 = 1
|
|
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) |