parse script fully command line

plot.py

@@ -26,14 +26,17 @@ def maternal_plot(exps, feature='multi', mode='mutations'):
 
 def subfigs(exps, dim1, dim2, mode='mutations', feature='multi'):
     count = 1
-    modes = ['mutations']
-    features = ['multi', 'energy']
+    modes = ['mutations', 'generation']
+    features = ['multi', 'energy', 'entropy']
     for mode in modes:
         for feature in features:
             plt.subplot(len(modes), len(features), count)
             count += 1
             for e in exps:
                 df = pd.read_csv(f"Data/{e}_df_{mode}_mean.csv")
+                if mode == 'generation':
+                    plt.plot(df[feature][:1000], label=f'{labels[e]}')
+                else:
                     plt.plot(df[feature], label=f'{labels[e]}')
             plt.ylabel(feature)
             plt.xlabel(mode)
@@ -41,7 +44,7 @@ def subfigs(exps, dim1, dim2, mode='mutations', feature='multi'):
     plt.tight_layout()
     plt.show()
 if __name__ == "__main__":
-    exps = ['orig', 'unp', 'nosel3', 'norm', 'beta01', 'lin2']
+    exps = [ 'beta01', 'unp', 'nosel3', 'norm', 'lin2']
     # maternal_plot(exps, feature='entropy')
     subfigs(exps, 2, 3)
     pass