Fix bad iloc indexing

jamiewan1989 · Jun 26, 2017 · b771642 · b771642
1 parent 24c72f1
commit b771642
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/missingno/missingno.py b/missingno/missingno.py
@@ -424,7 +424,7 @@ def heatmap(df, inline=True,
     ax0 = plt.subplot(gs[0])
 
     # Pre-processing: remove completely filled or completely empty variables.
-    df = df.iloc[[i for i, n in enumerate(np.var(df.isnull(), axis='rows')) if n > 0]]
+    df = df.iloc[:,[i for i, n in enumerate(np.var(df.isnull(), axis='rows')) if n > 0]]
 
     # Create and mask the correlation matrix.
     corr_mat = df.isnull().corr()