Skip to content

Instantly share code, notes, and snippets.

@balkian
Created October 25, 2019 14:58
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save balkian/41c51d1a6f6d4277c6cee5a92b8ba6d1 to your computer and use it in GitHub Desktop.
Save balkian/41c51d1a6f6d4277c6cee5a92b8ba6d1 to your computer and use it in GitHub Desktop.
def latex_max(data, bgcolor='yellow', fgcolor="black", ignore=['AvgContent',]):
'''
highlight the maximum in a Series or DataFrame
'''
txt = '{:.3f}'
attr = '\textbf{{{}}}'
attr_max = '\colorbox{{' + bgcolor + '}}{{\textcolor{{' + fgcolor + '}}{{{}}}}}'
# print(data)
is_max = data >= data.drop(ignore).max()
mx = data.drop(ignore).max()
real_mx = data.max()
attrs = []
for v in data:
a = txt.format(v)
if v == mx:
a = attr.format(a)
if v == real_mx:
a = attr_max.format(a)
attrs.append(a)
return pd.Series(attrs, index=data.index)
def to_latex(df, *args, **kwargs):
with pd.option_context('display.max_rows', None, 'display.max_columns', None, 'display.max_colwidth', -1):
print(df.apply(lambda x: latex_max(x, *args, **kwargs), axis=1).to_latex(escape=False))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment