base_sample.py 文件源码

def display_covariate_dist(self, covariate_list, save_file=None):
        '''
        '''

        n_covars = len(covariate_list)

        for covariate in covariate_list:
            g = sns.FacetGrid(self.data, col="arm_assignment")
            if len(self.data[covariate].unique())>2:
                g.map(sns.distplot, covariate, kde=False)
            else:
                g.map(sns.distplot, covariate, kde=False)
            if save_file:
                g.savefig(save_file, dpi=450)

        if save_file is None:
            sns.plt.show()