Added zeroes count for numerical feature stats

fbdesignpro · Feb 25, 2021 · 1aad4f3 · 1aad4f3
1 parent 41f0326
commit 1aad4f3
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 1 deletion.
diff --git a/sweetviz/series_analyzer.py b/sweetviz/series_analyzer.py
@@ -59,7 +59,7 @@ def add_series_base_stats_to_dict(series: pd.Series, counts: dict, updated_dict:
     base_stats = updated_dict["base_stats"]
     num_total = counts["num_rows_total"]
     try:
-        num_zeros = series[series == 0].sum()
+        num_zeros = series[series == 0].count()
     except TypeError:
         num_zeros = 0
     non_nan = counts["num_rows_with_data"]

diff --git a/sweetviz/templates/feature_summary_base_stats.html b/sweetviz/templates/feature_summary_base_stats.html
@@ -70,5 +70,42 @@
                 </div>
             {% endif %}
         </div>
+        {%  if feature_dict.type == FeatureType.TYPE_NUM %}
+            <div class="base-stats-row">
+            </div>
+            <div class="base-stats-row">
+                <div class="text-label color-normal pos-base-stats__label {{ "color-target-summary" if feature_dict.is_target }}">ZEROES:</div>
+                <div class="text-value color-source pos-base-stats__source">
+                    {% if feature_dict.base_stats.num_zeroes.number > 0: %}
+                        {{ feature_dict.base_stats.num_zeroes.number|fmt_int_commas }}
+                    {% else %}
+                        ---
+                    {% endif %}
+                </div>
+                <div class="text-value color-source pos-base-stats__source-perc">
+                    {% if feature_dict.base_stats.num_zeroes.number > 0: %}
+                        {{ feature_dict.base_stats.num_zeroes.perc|fmt_percent_parentheses }}
+                    {% else %}
+
+                    {% endif %}
+                </div>
+                {% if compare_dict is not none: %}
+                 <div class="text-value color-compare pos-base-stats__compare">
+                    {% if compare_dict.base_stats.num_zeroes.number > 0: %}
+                        {{ compare_dict.base_stats.num_zeroes.number|fmt_int_commas }}
+                    {% else %}
+                        ---
+                    {% endif %}
+                </div>
+                <div class="text-value color-compare pos-base-stats__compare-perc">
+                    {% if compare_dict.base_stats.num_zeroes.number > 0: %}
+                        {{ compare_dict.base_stats.num_zeroes.perc|fmt_percent_parentheses }}
+                    {% else %}
+
+                    {% endif %}
+                </div>
+                {% endif %}
+            </div>
+        {% endif %}
     </div>