'Totalplotter.py' toevoegen
This commit is contained in:
parent
c3fe14a6c5
commit
d96c52db24
61
Totalplotter.py
Normal file
61
Totalplotter.py
Normal file
@ -0,0 +1,61 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
Created on Sat Jun 26 13:27:00 2021
|
||||||
|
|
||||||
|
@author: D_Mik
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import pandas as pd
|
||||||
|
import formules
|
||||||
|
import numpy as np
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
|
||||||
|
# provide directory for formula parameters
|
||||||
|
rootdir = 'D:\AcA Mike Dijkhof\cwa files\Pt304_csv\Pt304_week_formula'
|
||||||
|
|
||||||
|
d = {}
|
||||||
|
X = list()
|
||||||
|
|
||||||
|
# import function parameters from previous scripts
|
||||||
|
|
||||||
|
for subdir, dirs, files in os.walk(rootdir):
|
||||||
|
for file in files:
|
||||||
|
print(file)
|
||||||
|
os.chdir(subdir)
|
||||||
|
|
||||||
|
name = file.replace('Formula_','')
|
||||||
|
name = name.replace('.csv','')
|
||||||
|
|
||||||
|
df = pd.DataFrame(pd.read_csv(file))
|
||||||
|
df = df.set_index('Name')
|
||||||
|
|
||||||
|
scale = df['ENMOmax'].max()
|
||||||
|
|
||||||
|
key = name
|
||||||
|
|
||||||
|
d[key] = df.loc['Mean']
|
||||||
|
X.append(scale)
|
||||||
|
|
||||||
|
# plot curves
|
||||||
|
plt.figure(dpi=720)
|
||||||
|
|
||||||
|
for key in d:
|
||||||
|
|
||||||
|
formula = d[key]
|
||||||
|
Xscale = np.arange(0, max(X), 5)
|
||||||
|
|
||||||
|
Y = formules.func(Xscale, formula.loc['a'], formula.loc['b'], formula.loc['c'] )
|
||||||
|
|
||||||
|
|
||||||
|
plt.ylim(0,1440)
|
||||||
|
plt.xlim(0,(max(X)+10))
|
||||||
|
plt.plot(Xscale, Y, label=key)
|
||||||
|
|
||||||
|
plt.legend()
|
||||||
|
plt.title('All 6 weeks plotter for 304') #Change to pt-number
|
||||||
|
plt.show()
|
||||||
|
|
||||||
|
plt.savefig(fname='Total_weeks_304.png')
|
||||||
|
|
||||||
|
|
Loading…
x
Reference in New Issue
Block a user