Fix post-processing and add __init__.py files

rl-institut · May 18, 2022 · a402e17 · a402e17
1 parent a2fd93e
commit a402e17
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 2 deletions.
diff --git a/ramp/__init__.py b/ramp/__init__.py
diff --git a/ramp/core/__init__.py b/ramp/core/__init__.py
diff --git a/ramp/input_files/__init__.py b/ramp/input_files/__init__.py
diff --git a/ramp/post_process/post_process.py b/ramp/post_process/post_process.py
@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 
 #%% Import required libraries
+import os.path
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
@@ -64,6 +66,9 @@ def Profile_series_plot(stoch_profiles_series):
 
 # Export Profiles
 
-def export_series(stoch_profiles_series, j):
+def export_series(stoch_profiles_series, j=None, fname=None):
     series_frame = pd.DataFrame(stoch_profiles_series)
-    series_frame.to_csv('results/output_file_%d.csv' % (j))
+    if j is not None:
+        series_frame.to_csv('results/output_file_%d.csv' % (j))
+    if fname is not None:
+        series_frame.to_csv(f'results/output_file_{os.path.split(fname)[-1]}.csv')
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+pandas
+numpy
+matplotlib