如何通过特定列的值创建数据框的两个子集？ [复制]

Question

所以我目前正在制作一个 python 机器学习程序，我想将数据框分成两部分。一个将包含等于 1 和另一个等于零的标签值。

这是迄今为止我的数据框的示例。

             Capital-loss  Hours-per-week  Label  
0             0              40              0  
1             0              13              1  
2             0              40              0  
3             0              40              1  
4             0              40              0  

我目前不确定如何根据标签的值进行拆分，或者最好的方法。

Answer 1

假设您的数据框名为df：

df_1 = df[df.Label==1]
df_0 = df[df.Label==0]

Answer 2

In [32]: import pandas as pd
    ...:
    ...: mainDf = pd.DataFrame()
    ...: mainDf['Type'] = [1,23,45,56,7,6]
    ...: mainDf['Dummy'] = [1,0,1,1,0,1]

In [33]: df1 = mainDf[mainDf.Dummy == 1]
    ...: df2 = mainDf[mainDf.Dummy == 0]

In [34]: df1
Out[34]:
   Type  Dummy
0     1      1
2    45      1
3    56      1
5     6      1

In [35]: df2
Out[35]:
   Type  Dummy
1    23      0
4     7      0