display.py 文件源码

def saveRegressionImage(self, event=None):
        if not self.regression_mode:
            tkMessageBox.showerror("Can't Save Image", "This feature only works for regression analysis currently.")
            return
        else:
            x = analysis.normalize_columns_separately(self.data, [self.x_label_name.get()]).T.tolist()[0]
            y = analysis.normalize_columns_separately(self.data, [self.y_label_name.get()]).T.tolist()[0]

            plt.plot(x, y, 'o')
            plt.plot([min(x), max(x)], [self.linreg_endpoints[0, 1], self.linreg_endpoints[1, 1]])
            plt.ylabel(self.y_label_name.get())
            plt.xlabel(self.x_label_name.get())

            plt.show()

    ### BEGIN PCA STUFF ###