predic_proba 输出是 dict 的列表：如何仅提取为 1 的概率？

Question

我有一个这样的字典列表：

[{'0':0.0, '1':1.0},
 {'0':0.3333, '1':0.666}]

这是 predict_proba 方法的输出。

现在我想传递一个数据框，概率仅为 '1'。

我在 scikit 中：

df['proba'] = model.proba[:,1]

[:,1]表示我想通过的概率为 1

我怎样才能对这个字典列表做同样的事情？

Answer 1

您可以使用here 中描述的列表推导获取值，并将所需的值分配给新列。

values_of_key = [a_dict[a_key] for a_dict in list_of_dicts]

这将为您提供仅包含所需值的列表。然后，您可以将此列表分配给新列。