def _helper_plot_heatmaps()

in aggregate.py [0:0]

• 43 lines of code
• 5 McCabe index (conditional complexity)

        def _helper_plot_heatmaps(data, save_dir):
            """Plot the results as heatmaps."""
            sns_plots = []
            data = get_transformer_data(transformer_data)(
                data,
                get_col_kwargs(data, **dict(row=row, col=col,
                                            x=x, y=y, **dflt_kwargs)),
            )

            for values, filename in gen_values_name(data):
                if data[values].isna().all():
                    logger.info(f"Skipping {filename} because all nan.")
                    continue

                pretty_data = self.prettify(data)
                pretty_kwargs = self.prettify_kwargs(
                    pretty_data, **dflt_kwargs)
                sns_plot = sns.FacetGrid(
                    pretty_data,
                    row=self.pretty_renamer[row],
                    col=self.pretty_renamer[col],
                    dropna=False,
                    sharex=True,
                    sharey=True,
                    aspect=1.5,
                    height=6,
                )
                sns_plot.map_dataframe(
                    _draw_heatmap,
                    self.pretty_renamer[x],
                    self.pretty_renamer[y],
                    values=self.pretty_renamer[values],
                    **pretty_kwargs,
                )
                sns_plot.fig.tight_layout()

                if self.is_return_plots:
                    logger.info(f"heatmap for {values}")
                    sns_plots.append(sns_plot)
                else:
                    sns_plot.fig.savefig(
                        os.path.join(save_dir, f"{self.prfx}{filename}.png"),
                        dpi=self.dpi,
                    )
                    plt.close(sns_plot.fig)

            if self.is_return_plots:
                return sns_plots