diff options
Improve plot output
Diffstat (limited to 'codegen_lbm.py')
-rw-r--r-- | codegen_lbm.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/codegen_lbm.py b/codegen_lbm.py index 8e7b0a7..29c2dad 100644 --- a/codegen_lbm.py +++ b/codegen_lbm.py @@ -189,8 +189,9 @@ class D2Q9_BGK_Lattice: for x in range(1,self.nX-1): density[y-1,x-1] = self.np_moments[0,self.idx(x,y)] + plt.figure(figsize=(10, 10)) plt.imshow(density, vmin=0.2, vmax=2.0, cmap=plt.get_cmap("seismic")) - plt.savefig("result/density_" + str(i) + ".png") + plt.savefig("result/density_" + str(i) + ".png", bbox_inches='tight', pad_inches=0) def MLUPS(cells, steps, time): |