def _violinplot( self, data: dataType, names: namesType, title: titleType = None, ax: matplotlib.axes.SubplotBase = None, ) -> matplotlib.figure.Figure: """For making violinplots.""" if ax is None: _, ax = plt.subplots() else: ax = ax figure = ax.get_figure() width = max(self.num_players / 3, 12) height = width / 2 spacing = 4 positions = spacing * arange(1, self.num_players + 1, 1) figure.set_size_inches(width, height) ax.violinplot( data, positions=positions, widths=spacing / 2, showmedians=True, showextrema=False, ) ax.set_xticks(positions) ax.set_xticklabels(names, rotation=90) ax.set_xlim([0, spacing * (self.num_players + 1)]) ax.tick_params(axis="both", which="both", labelsize=8) if title: ax.set_title(title) plt.tight_layout() return figure
def _violinplot( self, data: dataType, names: namesType, title: titleType = None, ax: matplotlib.axes.SubplotBase = None ) -> matplotlib.figure.Figure: """For making violinplots.""" if ax is None: _, ax = plt.subplots() else: ax = ax figure = ax.get_figure() width = max(self.nplayers / 3, 12) height = width / 2 spacing = 4 positions = spacing * arange(1, self.nplayers + 1, 1) figure.set_size_inches(width, height) ax.violinplot(data, positions=positions, widths=spacing / 2, showmedians=True, showextrema=False) ax.set_xticks(positions) ax.set_xticklabels(names, rotation=90) ax.set_xlim([0, spacing * (self.nplayers + 1)]) ax.tick_params(axis='both', which='both', labelsize=8) if title: ax.set_title(title) plt.tight_layout() return figure
def plot_2_2(ax: matplotlib.axes.SubplotBase, df_eff_k: pd.DataFrame, rows_set: np.ndarray) -> None: lty = ["--", "-", "--"] lwd = [1, 2, 1] mu, sigma = ClusteringUtils.calc_distribution(df_eff_k.loc[rows_set]) y = pd.DataFrame([mu - sigma, mu, mu + sigma]).transpose() for col in y.columns: ax.plot(df_eff_k.columns, y[col], color="red", linestyle=lty[col], linewidth=lwd[col]) ax.set_xlabel("k") ax.set_ylabel("effective k") ax.set_xlim(1, df_eff_k.columns.max()) ax.set_ylim(1, df_eff_k.columns.max()) abline_vals = np.array(ax.get_xlim()) ax.plot(abline_vals, abline_vals, color="grey", linestyle="--")