Skip to content
Snippets Groups Projects
save_table_better_simuls.py 1.84 KiB
Newer Older
vlabarre's avatar
vlabarre committed
from util_dataframe import df, df_proj

from util import tmp_dir


def formatter_R(v):
    if v % 1 == 0 or v >= 100:
        return f"{v:.0f}"
    else:
        return f"{v:.1f}"


def formatter_N(v):
    if v % 1 == 0:
        return f"{v:.0f}"
    elif v < 10:
        return f"{v:.2f}"
    else:
        return f"{v:.1f}"


formatters = {
    "N": formatter_N,
    "Rb": formatter_R,
    "k_max*eta": lambda v: f"{v:.2f}",
    "k_max*lambda": lambda v: f"{v:.2f}",
    "epsK2/epsK": lambda v: f"{v:.2f}",
    "Fh": lambda v: f"{v:.2e}",
    "R2": formatter_R,
    "R4": lambda v: f"{v:.2e}",
    "Re_lambda": formatter_R,
    "Re": formatter_R,
    "Lh/Lz": lambda v: f"{v:.0f}",
}
vlabarre's avatar
vlabarre committed

# print(df.columns)

df["Lh/Lz"] = df["nx"] / df["nz"]
columns = ["N", "Rb", "Lh/Lz"]
header = [r"$N$", r"$\R_i$", r"$L_h/L_z$"]
column_format = "|lll|"

df.to_latex(
    buf=tmp_dir / "table_params.tex",
    columns=columns,
    formatters=formatters,
    column_format=column_format,
    index=False,
    header=header,
    escape=False,
    caption=None,
    label=None,
)

vlabarre's avatar
vlabarre committed

Pierre Augier's avatar
Pierre Augier committed
    "k_max*eta",
    "Fh",
    "R2",
    # "Re", "R4", "epsK2/epsK", "k_max*lambda", "Re_lambda",
    # "Uh2", "epsK", "Gamma", "I_velocity", "I_dissipation",
column_format = "|" + len(columns) * "l" + "|"
df_tmp = df[columns]

header = ["$n_h$", r"$\kmax\eta$", r"$F_h$", r"$\mathcal{{R}}$"]
df_tmp.to_latex(
vlabarre's avatar
vlabarre committed
    buf=tmp_dir / "table_better_simuls.tex",
vlabarre's avatar
vlabarre committed
    formatters=formatters,
    column_format=column_format,
    index=False,
    header=header,
    escape=False,
    caption=None,
    label=None,

df_tmp = df_proj[columns]
df_tmp.to_latex(
vlabarre's avatar
vlabarre committed
    buf=tmp_dir / "table_better_simuls_proj.tex",
vlabarre's avatar
vlabarre committed
    formatters=formatters,
    column_format=column_format,
    index=False,
    header=header,
    escape=False,
    caption=None,
    label=None,
vlabarre's avatar
vlabarre committed
)