started using numpy like i should have to begin with...

plot_evaluations.py

@@ -1,8 +1,9 @@
 from numpy import load
+import numpy as np
 import matplotlib.pyplot as plt 
 # import matplotlib.axes
 
-filepath = "dqn_v2-5/evaluations.npz"
+filepath = "agents/dqn_v2-7/evaluations.npz"
 
 data = load(filepath)
 lst = data.files # data.files lists the keys that are available for data
@@ -11,7 +12,13 @@ lst = data.files # data.files lists the keys that are available for data
 
 # results and ep_lengths are 2d arrays, because each evaluation is 5 episodes long.
 # I want to plot the average of each evaluation.
+print(data["results"])
+print()
+print(np.delete(np.sort(data["results"]), 0, 1))
+# for i in range(len(data["results"])):
+#     print(np.average(data["results"][i]))
 
+'''
 # for each item in results, loop through the array and save the average
 avg_ep_result_arr = []
 for eval in data['results']:
@@ -49,4 +56,5 @@ plt.ylim(top=y_limit)
 # lineObjects = plt.plot(x, y)
 plt.legend(["avg ep result", "avg ep length"])
 plt.title("result and length over steps\nfilepath: " + filepath)
-plt.show()
+plt.show()
+'''