Move Python scripts
This commit is contained in:
27
python_scripts/pd_estimate.py
Normal file
27
python_scripts/pd_estimate.py
Normal file
@@ -0,0 +1,27 @@
|
||||
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")
|
||||
Reference in New Issue
Block a user