figure generationscripts

nicholscrawford · Jan 30, 2022 · 4cf4b16 · 4cf4b16
1 parent 3d7af1b
commit 4cf4b16
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 1 deletion.
diff --git a/cityFireFig.py b/cityFireFig.py
@@ -0,0 +1,9 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+grid = np.zeros((100,200))
+grid[49,57] = 0.9
+grid[46:54,125:142] = 0.1
+
+
+plt.imshow(grid,cmap=plt.get_cmap('gnuplot2'))
diff --git a/figures/burnTimes.png b/figures/burnTimes.png
diff --git a/figures/setupMatrix.png b/figures/setupMatrix.png
diff --git a/topModel.py b/topModel.py
@@ -22,4 +22,7 @@
             if testModel.state[j+46,k+125] != 0:
                 burnTimes[j,k] = i + 1
 
-fig, ax1 = plt.imshow(burnTimes,cmap=plt.get_cmap('inferno'))
+fig, ax = plt.subplots()
+im = plt.imshow(burnTimes,cmap=plt.get_cmap('RdBu'))
+fig.colorbar(im)
+plt.show()