dataframe 筛选_python—变量筛选

撰文  Yotsunoha

编辑  袁老师

这里我们使用logit回归和OLS(最小二乘回归)回归展示一下python的变量筛选过程。

from scipy import statsfrom statsmodels.formula.api import olsfrom statsmodels.stats.anova import anova_lmfrom statsmodels.stats.multicomp import pairwise_tukeyhsdimport warningswarnings.filterwarnings("ignore")import itertoolsdata = pd.read_csv("完整数据.csv")

读取数据，加载所需要的包。

columns = ['age', 'bicarbonate_max','rdw_max',  'phosphate_max', 'alf', 'af','stroke', 'map_mean', 'aki', 'calcium_max',       'sodium_max','phosphate_max','ck_max']column_P_2 = [] # 保存p值小于0.2的变量for column in columns:    sts, p_val = stats.ttest_ind(data.loc[(data.hosp_mort