python-3.x - 从 Pandas DataFrame 创建 StratifiedK-Folds

Question

假设我有一个df像这样的 PandasDataFrame：

|---------------------|------------------|
|         ID          |     Value        |
|---------------------|------------------|
|          12         |        32        |
|---------------------|------------------|
|          354        |        43        |
|---------------------|------------------|

值是随机的。

我想使用sklearn.model_selection.StratifiedKFold. 我想要的是添加另一列df[kfold]，该列将对应于通过拆分dataframe. 可能吗 ？

标签： python-3.xpandasscikit-learn