Coverage for mlair/helpers/tables.py: 100%

1import pandas as pd 

2import numpy as np 

3import os 

4 

5 

6def create_column_format_for_tex(df: pd.DataFrame) -> str: 

7 """ 

8 Creates column format for latex table based on the shape of a given DataFrame. 

9 

10 Calculates number of columns and uses 'c' as column position. First element is set to 'l', last to 'r' 

11 """ 

12 column_format = np.repeat('c', df.shape[1] + 1) 

13 column_format[0] = 'l' 

14 column_format[-1] = 'r' 

15 column_format = ''.join(column_format.tolist()) 

16 return column_format 

17 

18 

19def save_to_tex(path, filename, column_format, df, na_rep='---'): 

20 df.to_latex(os.path.join(path, filename), na_rep=na_rep, column_format=column_format) 

21 

22 

23def save_to_md(path, filename, df, mode="w", encoding='utf-8', tablefmt="github"): 

24 df.to_markdown(open(os.path.join(path, filename), mode=mode, encoding=encoding), tablefmt=tablefmt)