'Totalplotter.py' toevoegen

master
Dijkhof 1 year ago
parent c3fe14a6c5
commit d96c52db24
  1. 61
      Totalplotter.py

@ -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…
Cancel
Save