feature: simple memory optimization with a numpy parameter (ResidentM…

…ario#124)
Bacapatry · Nov 23, 2020 · a0de212 · a0de212
1 parent 1d8c590
commit a0de212
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/missingno/missingno.py b/missingno/missingno.py
@@ -40,7 +40,7 @@ def matrix(df,
 
     # z is the color-mask array, g is a NxNx3 matrix. Apply the z color-mask to set the RGB of each pixel.
     z = df.notnull().values
-    g = np.zeros((height, width, 3))
+    g = np.zeros((height, width, 3), dtype=np.float32)
 
     g[z < 0.5] = [1, 1, 1]
     g[z > 0.5] = color