ceres-sees-all/experiment-t
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update python

Browse source
This commit is contained in:
Ceres 2025-11-20 15:37:10 +00:00
parent cffbe828b3
commit a07fc681cc
Signed by: ceres-sees-all
GPG key ID: 9814758436430045
3 changed files with 84 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

7
energy.py
View file

@ -12,6 +12,7 @@ args=parser.parse_args()
columns = ["ID", "idx", "Mass", "Radius", "X", "Y", "Z", "vX", "vY", "vZ", "sX", "sY", "sZ", "Colour"]
Kenergies = []
Genergies = []
energies = []
time = []
for i in range (1,401):
@ -37,8 +38,9 @@ for i in range (1,401):
data["vel"] = data["distance2"]**0.5
data["KE"] = 0.5*data["Mass"]*data["vel2"]
data["GPE"] = (6.67e-11*1.89e27*data["Mass"])/(data["distance"])
# data["Energy"] = data["KE"]+data["GPE"]
# energy = data["Energy"].sum()
data["Energy"] = data["KE"]+data["GPE"]
energy = data["Energy"].sum()
energies.append(energy)
time.append(i)
KE = data["KE"].sum()
Kenergies.append(KE)
@ -49,5 +51,6 @@ ax=plt.axes()
# ax.set_yscale("log")
ax.plot(time, Kenergies, label="Kinetic")
ax.plot(time, Genergies, label="Gravitational")
ax.plot(time, energies, label="Total")
ax.legend()
plt.show()