Pandas 熊猫:如何使用predict_proba获得每个客户的概率

Pandas 熊猫:如何使用predict_proba获得每个客户的概率,pandas,Pandas,我使用xgboost和objective='binary:logistic'来计算每个客户是否会进行支出的概率。 在sklearn中使用predic_proba将打印0和1的两个概率,如: [[0.56651809 0.43348191] [0.15598162 0.84401838] [0.86852502 0.13147498]] 如何通过pandas插入每个客户ID,以获得如下信息: +----+------------+------------+ | ID | pro

我使用xgboost和objective='binary:logistic'来计算每个客户是否会进行支出的概率。 在sklearn中使用predic_proba将打印0和1的两个概率,如:

[[0.56651809 0.43348191]  
 [0.15598162 0.84401838]  
 [0.86852502 0.13147498]] 
如何通过pandas插入每个客户ID,以获得如下信息:

+----+------------+------------+
| ID |   prob_0   |   prob_1   |
+----+------------+------------+
|  1 | 0.56651809 | 0.43348191 |
|  2 | 0.15598162 | 0.84401838 |
|  3 | 0.86852502 | 0.13147498 |
+----+------------+------------+

可以使用DataFrame()创建表单

    list_data = [[0.56651809, 0.43348191],[0.15598162, 0.84401838],[0.86852502, 0.13147498]]
    columns = ['prob_0', 'prob_1']
    index = [1, 2, 3]
    pd.DataFrame(data = list_data, columns = columns, index= index)