如何从predict_proba输出的字典列表中提取类别为1的概率值

实现方法 #

你可以直接通过列表推导式遍历字典列表，提取每个字典中键为'1'的值直接赋值给DataFrame列即可，写法和scikit原生的操作一样简洁：

# 假设你的字典列表变量名为proba_list
df['proba'] = [item['1'] for item in proba_list]

如果要兼容部分字典可能不存在'1'键的异常场景，可以用get方法设置默认值：

# 取不到键'1'时默认返回0.0
df['proba'] = [item.get('1', 0.0) for item in proba_list]

你也可以选择先把字典列表转成DataFrame再取对应列，效果完全一致：

import pandas as pd
df['proba'] = pd.DataFrame(proba_list)['1']

内容的提问来源于stack exchange，提问作者Nyzky