'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