highlight_max.py

def latex_bold(text: str) -> str:
    """Format text in bold font using Latex."""
    return rf"\textbf{{{text}}}"


def highlight_max(
    data: pandas.Series,
    float_formatter: Callable[[float], str] = "{:2.2f}".format,
    highlighter: Callable[[str], str] = latex_bold,
) -> pandas.Series:
    """Highlight maximum value in each column."""
    is_max = data == data.max()
    data = data.apply(float_formatter).str.replace("nan", "")
    data[is_max] = data[is_max].apply(highlighter)
    return data