from .base import BaseTable
from .tableaxis import ConditionTableAxis
[docs]class ConditionTable(BaseTable):
def __init__(self, conds, *args, **kwargs):
self._dfc = conds # DataFrame of condition cases
super().__init__(*args, **kwargs)
"""
Add labels using x and y axes.
The default values should be assigned before adding labels.
"""
self.row_tick_fontsize = 5
self.column_tick_fontsize = 5
self._add_labels()
# end of def __init__
def _set_colors(self, colors):
"""Assign default color values, if it is not defined.
"""
self._set_default_color('cond_up_cell', 'blue')
self._set_default_color('cond_dn_cell', 'white')
def _create_axes(self):
super()._create_axes()
ax = self._axes['base']
self._axes['condition'] = ax
del self._axes['base']
def _create_tables(self):
self._tables = []
tb = ConditionTableAxis(self._axes['condition'],
self._dfc, self._colors)
tb.fontsize = 4
tb.linewidth = 0.5
self._tables.append(tb)
def _add_labels(self):
tb = self._tables[0]
tb.add_row_labels()
tb.add_column_labels()