convert-table.py

#%%
# merges all depth dose data into one single table
# first column: energy; rest of the columns: dose at the depths
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import os


directory = "data" # the folder containing all the depth-dose data
colnames=['X', 'Y', 'Z', 'Total Dose', 'Total2','Entries']
colnames_merged=['Energy','Position','Dose']
d =[]


def getEnergy(fileName):
    e =fileName.replace("-MEV-10000000-EVTS.txt","")
    return float(e)

for file in os.listdir(directory):
    filename = os.fsdecode(file)
    if filename.endswith(".txt"): #or filename.endswith(".py"): 
        #print(os.path.join(directory, filename))
        df = pd.read_csv(os.path.join(directory, filename),skiprows=3,header=0,names=colnames)   
        energy =getEnergy(filename)
        for i in range(0,299):
            d.append([energy,df['Z'][i],df['Total Dose'][i]])
        #plt.plot(df['Z'],df['Total Dose'])
        
        continue
    else:
        continue
df_merged = pd.DataFrame(d,columns=colnames_merged)
plt.plot(df_merged['Energy'],df_merged['Dose'])
#


# %%