pandasnotebooks-0.4.1.tar.gz

import logging import os import ipywidgets as widgets import numpy as np from IPython.display import display logger = logging.getLogger(__name__) class LatexWidget: def __init__( self, df, git_commit_ids, row_ids=None, output_file=None, include_index=False, float_format='{:,.3f}', use_multirow=True, use_multicolumn=True, column_names=None, column_includes=None, column_formats=None, overwrite_output_file=False, highlight_max_per_column=False, escape=False, index_column_format='l', na_rep='-', ): self.df = df self.output = widgets.Output() self.latex_out = widgets.Output(layout={'border': '1px solid grey'}) self.w_file_log_output = widgets.Output() self.w_include_index = widgets.Checkbox( description='include index', value=include_index ) self.w_index_column_format = widgets.Text( description='index col. format', value=index_column_format, disabled=not include_index, ) self.w_include_index.observe(self._update_index_column_format) self.w_float_format = widgets.Text( description='float format', value=float_format) self.w_na_rep = widgets.Text( description='NaN repr.', value=na_rep) self.w_use_multirow = widgets.Checkbox( description='use multirows', value=use_multirow) self.w_use_multicolumn = widgets.Checkbox( description='use multicolumns', value=use_multicolumn) self.w_highlight_max_per_column = widgets.Checkbox( description='print the max value of each column in bold', value=highlight_max_per_column) self.w_escape = widgets.Checkbox(description='Escape Latex', value=escape) self.w_output_file = widgets.Text(description='output file', value=output_file) self.w_draw_table_button = widgets.Button( description='draw latex table') self.w_write_file_button = widgets.Button( description='write to .tex file') self.w_overwrite_output_file = widgets.Checkbox( description='overwrite', value=overwrite_output_file ) self.w_cols = {} self.output_widgets = [] self.git_commit_ids = git_commit_ids self.row_ids = row_ids if column_names is None: column_names = df.columns if len(column_names) != len(df.columns): logger.error( "column_names doesn't have the correct length of " f'{len(df.columns)}' ) return if column_includes is None: column_includes = [True] * len(column_names) if len(column_includes) != len(df.columns): logger.error( "column_includes doesn't have the correct length of " f'{len(df.columns)}' ) return if column_formats is None: column_formats = ['c'] * len(column_names) if len(column_formats) != len(df.columns): logger.error( "column_formats doesn't have the correct length of " f'{len(df.columns)}' ) return for i, column in enumerate(self.df.columns): self.w_cols[column] = { 'name': widgets.Text( value=str(column_names[i] if column_names[i] else column), layout=widgets.Layout(width='140px'), ), 'sort_up': widgets.Button( icon='arrow-up', tooltip=str(column), layout=widgets.Layout(width='30px'), ), 'sort_down': widgets.Button( icon='arrow-down', tooltip=str(column), layout=widgets.Layout(width='30px'), ), 'export': widgets.Checkbox( value=column_includes[i], layout=widgets.Layout(width='140px') ), 'format': widgets.Text( value=column_formats[i], layout=widgets.Layout(width='30px') ), } self.output_widgets.append( widgets.HBox( [ widgets.Label( layout=widgets.Layout(min_width='140px'), value=str(column) ), self.w_cols[column]['export'], self.w_cols[column]['name'], self.w_cols[column]['sort_up'], self.w_cols[column]['sort_down'], self.w_cols[column]['format'], ] ) ) self.w_cols[column]['sort_up'].on_click(self._sort_up) self.w_cols[column]['sort_down'].on_click(self._sort_down) self.w_draw_table_button.on_click(self._draw_table) self.w_write_file_button.on_click(self._write_file) self._redraw() self.display() def _sort_up(self, button): cols = list(self.df.columns) old = cols.index(button.tooltip) if old == 0: return new = old - 1 cols[new], cols[old] = cols[old], cols[new] self.output_widgets[new], self.output_widgets[old] = ( self.output_widgets[old], self.output_widgets[new], ) self.df = self.df[cols] self._redraw() def _sort_down(self, button): cols = list(self.df.columns) old = cols.index(button.tooltip) if old == len(cols) - 1: return new = old + 1 cols[new], cols[old] = cols[old], cols[new] self.output_widgets[new], self.output_widgets[old] = ( self.output_widgets[old], self.output_widgets[new], ) self.df = self.df[cols] self._redraw() def _get_table_content(self): _columns = [ c for c in self.df.columns if self.w_cols[c]['export'].value ] selected_names = [self.w_cols[c]['name'].value for c in _columns] _formats = ''.join( [self.w_cols[c]['format'].value for c in _columns] ) if self.w_include_index.value: _formats = self.w_index_column_format.value + _formats result = '' if self.git_commit_ids is not None: result += '% This table was made with results from commit: ' result += str(self.git_commit_ids) + '\n' formatters = self._generate_formatters() if self.row_ids: r_min, r_max = min(self.row_ids), max(self.row_ids) result += '% This table contains results from rows:' result += '% ' + str(self.row_ids) + '\n' result += self.df.to_latex( columns=_columns, header=selected_names, column_format=_formats, index=self.w_include_index.value, float_format=self.nan_safe_float, multirow=self.w_use_multirow.value, multicolumn=self.w_use_multicolumn.value, formatters=formatters, escape=self.w_escape.value, ) return result def nan_safe_float(self, x): if np.isnan(x): return self.w_na_rep.value return self.w_float_format.value.format(x) def _generate_formatters(self): if self.w_highlight_max_per_column.value: return [self._generate_max_formatter(column) for column in self.df.columns] return None def