mirror of
https://github.com/Findus23/halo_comparison.git
synced 2024-09-19 16:03:50 +02:00
36 lines
928 B
Python
36 lines
928 B
Python
"""
|
|
Looks like this file takes one snapshot (line 26) and visualises specified halos (line 28) as a 2d-projection. Might not be accurate, hasn't been changed in a long time.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import pandas as pd
|
|
from matplotlib.axes import Axes
|
|
from matplotlib.colors import LogNorm
|
|
from matplotlib.figure import Figure
|
|
|
|
from paths import base_dir
|
|
from readfiles import read_file
|
|
|
|
|
|
def filter_for_2d(df: pd.DataFrame, group: int):
|
|
xs = df_ref.X
|
|
ys = df_ref.Y
|
|
return np.array([xs, ys]).T
|
|
|
|
|
|
reference_dir = Path(base_dir / f"shannon_512_100")
|
|
df_ref, _ = read_file(reference_dir / "output_0004.hdf5")
|
|
|
|
df = df_ref.loc[df_ref["FOFGroupIDs"] == 1]
|
|
df2 = df_ref.loc[df_ref["FOFGroupIDs"] == 4]
|
|
|
|
fig: Figure = plt.figure()
|
|
ax: Axes = fig.gca()
|
|
|
|
# ax.hist2d(df.X, df.Y, bins=500, norm=LogNorm())
|
|
ax.hist2d(df2.X, df2.Y, bins=1000, norm=LogNorm())
|
|
|
|
plt.show()
|