def plot_scatter()

in compert/plotting.py [0:0]

• 59 lines of code
• 7 McCabe index (conditional complexity)

    def plot_scatter(
        self,
        df,
        x_axis,
        y_axis,
        hue=None,
        size=None,
        style=None,
        figsize=(4.5, 4.5),
        title=None,
        palette=None,
        filename=None,
        alpha=.75,
        sizes=(30, 90),
        text_dict=None,
        postfix='',
        fontsize=14):

        sns.set_style("white")

        if (filename is None) and not (self.fileprefix is None):
            filename = f'{self.fileprefix}_scatter{postfix}.png'

        if fontsize is None:
            fontsize = self.fontsize

        fig = plt.figure(figsize=figsize)
        ax = plt.gca()
        sns.scatterplot(
                    x=x_axis,
                    y=y_axis,
                    hue=hue,
                    style=style,
                    size=size,
                    sizes=sizes,
                    alpha=alpha,
                    palette=palette,
                    data=df)

        ax.legend_.remove()
        ax.set_xlabel(x_axis, fontsize=fontsize)
        ax.set_ylabel(y_axis, fontsize=fontsize)
        ax.xaxis.set_tick_params(labelsize=fontsize)
        ax.yaxis.set_tick_params(labelsize=fontsize)
        ax.set_title(title)
        if not (text_dict is None):
            texts = []
            for label in text_dict.keys():
                texts.append(
                    ax.text(
                        text_dict[label][0],
                        text_dict[label][1],
                        label,
                        fontsize=fontsize
                    )
                )

            adjust_text(
                texts,
                arrowprops=dict(arrowstyle='-', color='black', lw=0.1),
                ax=ax
            )

        plt.tight_layout()

        if filename:
            save_to_file(fig, filename)