OWBoxPlot: Plot missing values

Author: PrimozGodec

Orange/statistics/contingency.py

@@ -125,6 +125,8 @@ def __getitem__(self, index):
             if isinstance(index[1], str):
                 index = (index[0], self.col_variable.to_val(index[1]))
         result = super().__getitem__(index)
+        if isinstance(index, int) or len(index) == 1:
+            result.unknowns = self.unknowns[index]
         if result.strides:
             result.col_variable = self.col_variable
             result.row_variable = self.row_variable

Orange/statistics/distribution.py

@@ -211,6 +211,11 @@ def max(self):
         if self.variable.ordered:
             return self.variable.values[-1]
 
+    def sum(self, *args, **kwargs):
+        res = super().sum(*args, **kwargs)
+        res.unknowns = self.unknowns
+        return res
+
 
 class Continuous(Distribution):
     def __new__(cls, dat, variable=None, unknowns=None):

Orange/widgets/visualize/owboxplot.py

@@ -987,15 +987,18 @@ def line(x0, y0, x1, y1, *args):
 
     def strudel(self, dist, group_val_index=None):
         attr = self.attribute
-        ss = np.sum(dist)
+        ss = np.sum(dist) + dist.unknowns
         box = []
         if ss < 1e-6:
             cond = [FilterDiscrete(attr, None)]
             if group_val_index is not None:
                 cond.append(FilterDiscrete(self.group_var, [group_val_index]))
             box.append(FilterGraphicsRectItem(cond, 0, -10, 1, 10))
         cum = 0
-        for i, v in enumerate(dist):
+        values = attr.values + ["Missing values"]
+        colors = np.vstack((attr.colors, [128, 128, 128]))
+        dist_unk = np.append(dist, dist.unknowns)
+        for i, v in enumerate(dist_unk):
             if v < 1e-6:
                 continue
             if self.stretched:
@@ -1005,15 +1008,16 @@ def strudel(self, dist, group_val_index=None):
             if group_val_index is not None:
                 cond.append(FilterDiscrete(self.group_var, [group_val_index]))
             rect = FilterGraphicsRectItem(cond, cum + 1, -6, v - 2, 12)
-            rect.setBrush(QBrush(QColor(*attr.colors[i])))
+            rect.setBrush(QBrush(QColor(*colors[i])))
             rect.setPen(QPen(Qt.NoPen))
             if self.stretched:
-                tooltip = "{}: {:.2f}%".format(attr.values[i],
-                                               100 * dist[i] / sum(dist))
+                tooltip = "{}: {:.2f}%".format(
+                    values[i],
+                    100 * dist_unk[i] / sum(dist_unk))
             else:
-                tooltip = "{}: {}".format(attr.values[i], int(dist[i]))
+                tooltip = "{}: {}".format(values[i], int(dist_unk[i]))
             rect.setToolTip(tooltip)
-            text = QGraphicsTextItem(attr.values[i])
+            text = QGraphicsTextItem(values[i])
             box.append(rect)
             box.append(text)
             cum += v