def HighCol(col, dct={"z":"white"}):
    """
    Use this as an argument in dataframe.style.apply,
    to highlight columns specified by keys of dictionary dct.
    """

    if col.name in dct.keys():
        return ["background-color: {}".format(dct[col.name])]*len(col)
    return [''] * len(col)