def _payoff_heatmap( self, data: dataType, names: namesType, title: titleType = None, ax: matplotlib.axes.SubplotBase = None ) -> matplotlib.figure.Figure: """Generic heatmap plot""" if not self.matplotlib_installed: return None if ax is None: _, ax = plt.subplots() else: ax = ax figure = ax.get_figure() width = max(self.nplayers / 4, 12) height = width figure.set_size_inches(width, height) cmap = default_cmap() mat = ax.matshow(data, cmap=cmap) plt.xticks(range(self.result_set.nplayers)) plt.yticks(range(self.result_set.nplayers)) ax.set_xticklabels(names, rotation=90) ax.set_yticklabels(names) plt.tick_params(axis='both', which='both', labelsize=16) if title: plt.xlabel(title) # Make the colorbar match up with the plot divider = make_axes_locatable(plt.gca()) cax = divider.append_axes("right", "5%", pad="3%") plt.colorbar(mat, cax=cax) return figure
def _payoff_heatmap( self, data: dataType, names: namesType, title: titleType = None, ax: matplotlib.axes.SubplotBase = None ) -> matplotlib.figure.Figure: """Generic heatmap plot""" if ax is None: _, ax = plt.subplots() else: ax = ax figure = ax.get_figure() width = max(self.nplayers / 4, 12) height = width figure.set_size_inches(width, height) matplotlib_version = matplotlib.__version__ cmap = default_cmap(matplotlib_version) mat = ax.matshow(data, cmap=cmap) ax.set_xticks(range(self.result_set.nplayers)) ax.set_yticks(range(self.result_set.nplayers)) ax.set_xticklabels(names, rotation=90) ax.set_yticklabels(names) ax.tick_params(axis='both', which='both', labelsize=16) if title: ax.set_xlabel(title) figure.colorbar(mat, ax=ax) plt.tight_layout() return figure
def _payoff_heatmap( self, data: dataType, names: namesType, title: titleType = None, ax: matplotlib.axes.SubplotBase = None, ) -> matplotlib.figure.Figure: """Generic heatmap plot""" if ax is None: _, ax = plt.subplots() else: ax = ax figure = ax.get_figure() width = max(self.num_players / 4, 12) height = width figure.set_size_inches(width, height) matplotlib_version = matplotlib.__version__ cmap = default_cmap(matplotlib_version) mat = ax.matshow(data, cmap=cmap) ax.set_xticks(range(self.result_set.num_players)) ax.set_yticks(range(self.result_set.num_players)) ax.set_xticklabels(names, rotation=90) ax.set_yticklabels(names) ax.tick_params(axis="both", which="both", labelsize=16) if title: ax.set_xlabel(title) figure.colorbar(mat, ax=ax) plt.tight_layout() return figure