def test_maybe_numeric_slice(self): df = DataFrame({"A": [1, 2], "B": ["c", "d"], "C": [True, False]}) result = maybe_numeric_slice(df, slice_=None) expected = pd.IndexSlice[:, ["A"]] assert result == expected result = maybe_numeric_slice(df, None, include_bool=True) expected = pd.IndexSlice[:, ["A", "C"]] result = maybe_numeric_slice(df, [1]) expected = [1] assert result == expected
def get_heatmap(self, cmap, sub=None, text_color_threshold=0.408): if sub is None: sub = self.subset cmap = self.min_max_scaler(sub).applymap( lambda x: self.get_heatmap_color(x, cmap, text_color_threshold)) sub = maybe_numeric_slice(self.data, sub) sub = non_reducing_slice(sub) c_mapping = lambda x: cmap.loc[sub] self.apply(c_mapping, subset=sub, axis=None) return self