プログラム一覧に戻る

import numpy as np  # データ読み込み機能などを使えるようにする
import matplotlib.pyplot as plt  # グラフプロット機能を使えるようにする

data = np.loadtxt("d4r1.txt")  # データファイルの読み込み

plt.xlabel("time t")  # 横軸の名前
plt.ylabel("total kinetic energy")  # 縦軸の名前

N = 4  # 粒子数
kinetic_ene = 0.5 * np.sum(data[:, 2 * N + 1 :] ** 2, axis=1)  # 各時刻での全系の運動エネルギー

plt.plot(data[:, 0], kinetic_ene, linestyle="", color="green", marker="o")

plt.show()  # グラフを表示