我将训练模型的信息存储在一个DataFrame中:
score timeLogisticRegression 1.0 0.257812RidgeClassifier 1.0 0.283841RandomForestClassifier 1.0 0.569159GradientBoostingClassifier 1.0 9.408432
我需要找到得分最高且时间最短的模型的索引(名称)。我该怎么做?
回答:
使用sort_values
函数,一侧升序,另一侧降序排序:
>>> df.sort_values(["score", "time"], ascending=[False, True]).index[0]"LogisticRegression"
然后获取结果中索引的第0个值。