From cd9c16ba4007c8f56708c7ebad0f78ba3ad5274d Mon Sep 17 00:00:00 2001
From: Simon Collignon <scllgnn@gmail.com>
Date: Tue, 20 Apr 2021 15:34:01 +0200
Subject: [PATCH] imshow instead of contourf
---
main.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/main.py b/main.py
index 8681d78..1d8eca0 100644
--- a/main.py
+++ b/main.py
@@ -39,7 +39,7 @@ def cahn_hilliard_spectral(it, c):
# wavenumbers
kr = power(2 * pi * fft.rfftfreq(nn, d=dx), 2)
kc = power(2 * pi * fft.fftfreq(nn, d=dx), 2)
- kk = array([kr,]*nn) + array([kc,]*int((nn/2)+1)) .transpose()
+ kk = array([kr,]*nn) + array([kc,]*int((nn/2)+1)).transpose()
for _ in range(it):
@@ -61,8 +61,8 @@ if __name__ == "__main__":
fig, (ax1, ax2) = plt.subplots(1, 2)
fig.suptitle(f"Cahn-Hilliard integration for {it} iterations ")
- ax1.contourf(xy, xy, c1, cmap="jet")
+ ax1.imshow(c1, cmap="jet", vmin=-1, vmax=1)
ax1.set_title("Finite differences method")
- ax2.contourf(xy, xy, c2 , cmap="jet")
+ ax2.imshow(c2, cmap="jet", vmin=-1, vmax=1)
ax2.set_title("Spectral method")
plt.show()
--
GitLab