1
0
Fork 0
mirror of https://github.com/Findus23/halo_comparison.git synced 2024-09-19 16:03:50 +02:00
halo_comparison/sizes.py

34 lines
691 B
Python

from sys import argv
import pandas as pd
from matplotlib import pyplot as plt
from matplotlib.axes import Axes
from matplotlib.figure import Figure
file = argv[1]
df = pd.read_csv(file)
print(df)
# df = df.iloc
fig: Figure = plt.figure()
ax: Axes = fig.gca()
# ax.scatter(df["ref_sizes"], df["comp_sizes"], s=1, alpha=.3)
ax.scatter(df["ref_masses"], df["comp_masses"], s=1, alpha=.3)
ax.set_xscale("log")
ax.set_xlabel("reference size")
ax.set_ylabel("comparison size")
ax.set_yscale("log")
fig2: Figure = plt.figure()
ax2: Axes = fig2.gca()
ax2.hist(df["distances"][df["distances"] < 50], bins=100)
ax2.set_ylabel("distance")
for a in [ax, ax2]:
a.set_title(file)
plt.show()