plot_matrix.py 文件源码

def __init__(self, data, **kwargs):
        # Settings
        self.show_ticks = kwargs.get("show_ticks", False)
        self.show_values = kwargs.get("show_values", False)
        self.show = kwargs.get("show", False)
        self.labels = kwargs.get("labels", None)

        # Setup plot
        self.rows, self.cols = data.shape
        self.fig = plt.figure()
        self.plt_ax = self.fig.add_subplot(111)
        self.cov_ax = self.plt_ax.matshow(np.array(data))

        # Covariance matrix labels
        self.label_values = self._add_data_labels(data)
        self._add_axis_labels(data)

        # Color bar
        self.color_bar = self.fig.colorbar(self.cov_ax)

        # Show plot
        if self.show:
            plt.show(block=False)