Add z-axis label to differential mesh 3D plots

tryuan99 · Jun 7, 2024 · 45abeba · 45abeba
1 parent 75edc55
commit 45abeba
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/simulation/differential_mesh/differential_mesh_analysis_main.py b/simulation/differential_mesh/differential_mesh_analysis_main.py
@@ -36,6 +36,7 @@ def plot_standard_error_sweep(standard_errors: str) -> None:
     )
     ax.set_xlabel("Number of rows")
     ax.set_ylabel("Number of columns")
+    ax.set_zlabel("Standard error")
     ax.view_init(30, -45)
     plt.colorbar(surf)
     plt.show()
@@ -55,6 +56,7 @@ def plot_standard_error_sweep(standard_errors: str) -> None:
     )
     ax.set_xlabel("Number of rows")
     ax.set_ylabel("Number of columns")
+    ax.set_zlabel("Squared standard error")
     ax.view_init(30, -45)
     plt.colorbar(surf)
     plt.show()

diff --git a/simulation/differential_mesh/differential_mesh_iterative_solver_main.py b/simulation/differential_mesh/differential_mesh_iterative_solver_main.py
@@ -86,6 +86,7 @@ def plot_num_iterations(num_iterations_per_grid: str) -> None:
     )
     ax.set_xlabel("Number of rows")
     ax.set_ylabel("Number of columns")
+    ax.set_zlabel("Mean number of iterations")
     ax.view_init(30, -45)
     plt.colorbar(surf)
     plt.show()

diff --git a/simulation/differential_mesh/differential_mesh_simulator_main.py b/simulation/differential_mesh/differential_mesh_simulator_main.py
@@ -91,6 +91,7 @@ def simulate_standard_error(solver: DifferentialMeshSolver, num_rows: int,
     )
     ax.set_xlabel("Row")
     ax.set_ylabel("Column")
+    ax.set_zlabel("Standard error")
     ax.view_init(30, -45)
     plt.colorbar(surf)
     plt.show()
@@ -141,6 +142,7 @@ def simulate_standard_error_sweep(solver: DifferentialMeshSolver,
     )
     ax.set_xlabel("Number of rows")
     ax.set_ylabel("Number of columns")
+    ax.set_zlabel("Standard error")
     ax.view_init(30, -45)
     plt.colorbar(surf)
     plt.show()

diff --git a/simulation/differential_mesh/differential_mesh_standard_error_main.py b/simulation/differential_mesh/differential_mesh_standard_error_main.py
@@ -45,6 +45,7 @@ def plot_standard_error(num_rows: int, num_cols: int) -> None:
     )
     ax.set_xlabel("Row")
     ax.set_ylabel("Column")
+    ax.set_zlabel("Standard error")
     ax.view_init(30, -45)
     plt.colorbar(surf)
     plt.show()