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

speed up similarity calculation

Browse source
This commit is contained in:
Lukas Winkler 2022-07-20 15:21:35 +02:00
parent 22c537aeec
commit 15b43bb7c1
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
compare_halos.py
View file

@ -235,9 +235,12 @@ def compare_halo_resolutions(
halo_size = len(particle_ids_in_comp_halo)
# df = particles_in_comp_halo.join(halo_particles, how="inner", rsuffix="ref")
shared_particles = particle_ids_in_comp_halo.intersection(halo_particle_ids)
union_particles = particle_ids_in_comp_halo.union(halo_particle_ids)
# union_particles = particle_ids_in_comp_halo.union(halo_particle_ids)
similarity = len(shared_particles) / len(union_particles)
# similarity = len(shared_particles) / len(union_particles)
similarity = len(shared_particles) / (
len(halo_particle_ids) + len(particle_ids_in_comp_halo) - len(shared_particles))
# assert similarity_orig == similarity
# print(shared_size)
# if not similarity:
# continue
@ -289,7 +292,7 @@ def compare_halo_resolutions(
pl.show()
if single:
break
counters.num_matches=len(compared_halos)
counters.num_matches = len(compared_halos)
df = pd.concat(compared_halos, axis=1).T
print(df)
print(f"saving to {outfile}")