'Totalplotter.py' toevoegen
This commit is contained in:
		
							
								
								
									
										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') 
 | 
				
			||||||
 | 
					    
 | 
				
			||||||
 | 
					        
 | 
				
			||||||
		Reference in New Issue
	
	Block a user