From bed10d85fe3b758c67bd024b3c18e9a5d905bea9 Mon Sep 17 00:00:00 2001 From: Bertrand Benjamin Date: Fri, 25 Jun 2021 07:57:00 +0200 Subject: [PATCH] Fix: rename grand_rectangle -> grand_carre --- grand_rectangle.py => grand_carre.py | 38 +++++++++++++++------------- 1 file changed, 21 insertions(+), 17 deletions(-) rename grand_rectangle.py => grand_carre.py (65%) diff --git a/grand_rectangle.py b/grand_carre.py similarity index 65% rename from grand_rectangle.py rename to grand_carre.py index 4cbfca6..0806f70 100644 --- a/grand_rectangle.py +++ b/grand_carre.py @@ -23,6 +23,26 @@ def trim_axs(axs, N): return axs[:N] +def draw_image(grid, ax): + hight = len(grid) + lenght = len(grid[0]) + ax.tick_params(left=False, bottom=False, labelleft=False, labelbottom=False) + ax.imshow(grid, cmap="binary") + ax.hlines( + y=np.arange(0, hight) + 0.5, + xmin=np.full(lenght, 0) - 0.5, + xmax=np.full(lenght, hight) - 0.5, + color="gray", + ) + ax.vlines( + x=np.arange(0, lenght) + 0.5, + ymin=np.full(lenght, 0) - 0.5, + ymax=np.full(lenght, hight) - 0.5, + color="gray", + ) + return ax + + def grid_to_image(grids, filename): """ """ plt.clf() @@ -31,23 +51,7 @@ def grid_to_image(grids, filename): axs = plt.figure(constrained_layout=True).subplots(n_rows, n_cols) axs = trim_axs(axs, len(grids)) for ax, grid in zip(axs, grids): - # for i, grid in enumerate(grids): - hight = len(grid) - lenght = len(grid[0]) - ax.tick_params(left=False, bottom=False, labelleft=False, labelbottom=False) - ax.imshow(grid, cmap="binary") - ax.hlines( - y=np.arange(0, hight) + 0.5, - xmin=np.full(lenght, 0) - 0.5, - xmax=np.full(lenght, hight) - 0.5, - color="gray", - ) - ax.vlines( - x=np.arange(0, lenght) + 0.5, - ymin=np.full(lenght, 0) - 0.5, - ymax=np.full(lenght, hight) - 0.5, - color="gray", - ) + draw_image(grid, ax) plt.savefig(filename)