import re
def text_to_latex(text):
# lazy funciton to fix symbols for LaTeX
if isinstance(text, str):
if text == "nan":
return ""
else:
if "%" in text:
text = re.sub(r"%", r"\%", text)
if "&" in text:
text = re.sub("&", r"\&", text)
if "μ" in text:
text = text.replace("μ", r"$\mu$")
if "" in text:
text = text.replace("Δ", r"$\delta$")
if "_" in text:
text = text.replace("_", r"\_")
if "#" in text:
text = text.replace("#", r"\#")
if "\n" in text:
text = text.replace("\n", " \\newline ")
# if "Unnamed" in text:
# text = " "
return text
def fix_multi_row_for_longtable(latex):
# see multirow package documentation for how this works
lines = latex.split('\n')
body_start = False
last_starred = -1
for i, l in enumerate(lines):
if "endlastfoot" in l:
body_start = True
continue
if body_start:
if 'multirow' in l or 'end{longtable}' in l:
if last_starred != -1:
lines[last_starred] = lines[last_starred].replace(r'\\*',r'\\')
if r'\\' in l:
lines[i] = lines[i].replace(r'\\',r'\\*')
last_starred = i
return '\n'.join(lines)
def passfail(c):
# applies color to cells with "pass" or "fail" in them.
if c.lower() == 'pass':
color = "light-green"
elif c.lower() == 'fail':
color = 'light-red'
else:
return None
return f"background-color:{color};"
def color_fract(c, denom):
if isinstance(c, str):
pattern_search = fr"([^{denom}]/{denom})"
pattern_replace = r'\\textcolor{red}{\1}'
sub = re.sub(pattern_search, pattern_replace, str(c))
else:
return c
return sub
def bold(c):
return f"textbf:--rwrap;"
# list of dataframes going to convert to LateX code
df_to_latex_tbl_list = [descr_stats_by_unit, descr_stats_by_sensor, trial_score_display, unit_score_display, sensor_score_display]
captions = ["Sensor performance descriptive statistics by unit.",
"Sensor performance descriptive statistics by sensor.",
"Sensor perforamnce scoring by trail.",
"Sensor perforamnce scoring by unit.",
"Sensor perforamnce scoring by trail."]
latex_tables = []
for df, caption in zip(df_to_latex_tbl_list, captions):
tdf = df.copy()
# apply replacements for certain symbols for LaTeX
tdf = tdf.applymap(text_to_latex)
tdf.columns = pd.Series(tdf.columns).apply(text_to_latex)
tdf.index.names = pd.Series(tdf.index.names).apply(text_to_latex)
# additional colors for scores (easier avoid styler for this)
tdf = tdf.applymap(color_fract,denom=3)
tdf = tdf.applymap(color_fract,denom=9)
# styler
styler = tdf.style
styler.applymap_index(bold, axis="columns")
styler.applymap_index(bold, axis="index")
styler.format({'mean' : '{:.2f}', 'Trials \%' : '{:.2f}', 'Units \%' : '{:.2f}' })
if "Score" in list(tdf.columns):
styler.applymap(passfail, subset="Score")
# I cannot figure out how to apply style to multi index names, so I am doing so "manually"
tdf.index.names = pd.Series(tdf.index.names).apply(lambda v: rf'\textbf{{{v}}}')
tbl = styler.to_latex(environment="longtable", position="H", caption=caption, hrules=True, convert_css=True)
tbl = fix_multi_row_for_longtable(tbl)
# apply some additional color
latex_tables.append(tbl)
