From 5742a42366ff4402c6ca7a91eceeeb45e29bed7c Mon Sep 17 00:00:00 2001
From: linesd <david.lines.dl@gmail.com>
Date: Tue, 8 Oct 2019 00:22:22 +0100
Subject: [PATCH] added 2d plot function

---
 utils/plots.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 utils/plots.py

diff --git a/utils/plots.py b/utils/plots.py
new file mode 100644
index 0000000..5f71235
--- /dev/null
+++ b/utils/plots.py
@@ -0,0 +1,22 @@
+import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
+
+def plot_2d_function_convergence(X, Y, Z, convergence, path=None):
+
+    # plot data
+    fig = plt.figure(figsize=(8,6))
+    ax = plt.axes(projection='3d')
+
+    ax.view_init(elev=45., azim=45)
+    ax.plot(convergence[:,1], convergence[:,2], convergence[:,0], '-r*', markersize=12)
+    ax.scatter(1, 1, 0, s=100)
+    ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='viridis',
+                    edgecolor='none', alpha=0.3, antialiased=False)
+    ax.set_xlabel('$X_{1}$', fontsize=20)
+    ax.set_ylabel('$X_{2}$', fontsize=20)
+    ax.set_zlabel('$f(\mathbf{X})$', fontsize=20)
+
+    if path is not None:
+        plt.savefig(path, dpi=200)
+
+    plt.show()
\ No newline at end of file