Loading...
Searching...
No Matches
pd_estimate.py
import os
from pathlib import Path
import matplotlib.pyplot as plt
import numpy as np
def plot(infile, outfile):
if not (outdir := Path(outfile).parent).exists():
os.makedirs(outdir)
figure1, ax1 = plt.subplots()
arr = []
with open(infile) as f:
lines = f.readlines()
for line in lines:
vals = line.strip().split(",")
arr.append(float(vals[0]))
ax1.hist(arr, np.arange(min(arr), max(arr) + 0.02, 0.02), density=True, ec="black")
figure1.savefig(outfile)
if __name__ == "__main__":
plot("output/pd_estimate/estimate_1_0.txt", "../latex/images/pd_estimate_1_0.pdf")
plot("output/pd_estimate/estimate_2_4.txt", "../latex/images/pd_estimate_2_4.pdf")