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)