如何将列表变量及对应循环计算的系数添加至DataFrame
将变量与对应系数存入DataFrame的方法
方法一:直接用字典构造(推荐)
这是最简洁高效的方式,直接把变量名和系数数组传入pd.DataFrame:
import pandas as pd variables = ['positional_versatility', 'weak_foot', 'skill_moves'] # 假设model.coef_是你的模型系数数组,如array([0.0041, -0.0421, 0.0010]) coef_df = pd.DataFrame({ '变量名称': variables, '系数值': model.coef_ }) # 查看结果 print(coef_df)
输出结果:
变量名称 系数值 0 positional_versatility 0.0041 1 weak_foot -0.0421 2 skill_moves 0.0010
方法二:通过循环构建列表后转换
如果需要在循环中对变量或系数做额外处理,可以先收集数据再转成DataFrame:
import pandas as pd variables = ['positional_versatility', 'weak_foot', 'skill_moves'] coef_list = [] for v, c in zip(variables, model.coef_): # 这里可以添加额外处理逻辑,比如格式化系数 coef_list.append({ '变量名称': v, '系数值': round(c, 4) # 示例:保留4位小数 }) coef_df = pd.DataFrame(coef_list)
内容的提问来源于stack exchange,提问作者Tom Bomer
相关产品推荐
相关产品推荐

