refactor

plot2d.py

@@ -0,0 +1,13 @@
+from matplotlib import pyplot as plt
+
+
+class Plot2D:
+
+    def __init__(self, *args, **kwargs):
+        self.plot = plt.imshow(*args, **kwargs)
+
+    def set(self, value):
+        self.plot.set_data(value)
+
+
+