python - 如何根据分类变量划分数据框？

Question

我有一个数据集，其中一些人的信用卡申请被接受，而其他人则被拒绝。

我想把数据集分成两个数据集；一张接受所有信用卡(card='yes')，另一张拒绝所有信用卡(card='no')。

数据集如下图所示:

我该怎么做？

Answer 1

一个选项是在 dict 理解中执行 groupby 操作。这具有处理任意数量类别的额外好处。

dfs_by_card = {
   accepted: sub_df
   for accepted, sub_df in credit5.groupby("card")
}