-
Notifications
You must be signed in to change notification settings - Fork 0
/
get_quantiles_width.py
41 lines (31 loc) · 1.05 KB
/
get_quantiles_width.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# -*- coding: utf-8 -*-
"""Module one liner
This module does what....
Example usage:
"""
import pandas as pd
import tqdm
import pandas as pd
import plot_corner_weighted_with_kick
FILE = "/fred/oz117/avajpeyi/projects/phase-marginalisation-test/jobs/out_hundred_injections_gstar/out_injection_{num}/result/injection_{num}_0_posterior_samples_with_kicks.dat"
def main():
widths = []
truths = pd.read_csv('datafiles/injections.csv')
for index, truth in tqdm.tqdm(truths.iterrows(), total=len(truths)):
f = FILE.format(num=index)
kick_mean = truth['remnant_kick_mag']
kick_sigma = 50
samples = plot_corner_weighted_with_kick.Samples(
samples_csv=f,
kick_mean=kick_mean,
kick_sigma=kick_sigma,
truths=truth.to_dict()
)
try:
widths.append(samples.get_quantiles_width('chi_p'))
except Exception as e:
print(f"Error: {e}")
df = pd.DataFrame(widths)
df.to_csv("quantiles_widths.csv")
if __name__ == "__main__":
main()