python - pandas:基于行值的列样本

例如我有这个数据框：

 d = {'data_1' : pd.Series([1., 2., 3., 4., 1], index=['a', 'b', 'c', 'd', 'class']),
  'data_2' : pd.Series([5., 6., 7., 8., 0], index=['a', 'b', 'c', 'd', 'class']),
  'data_3' : pd.Series([9., 10., 11., 12., 1], index=['a', 'b', 'c', 'd', 'class'])}
df = pd.DataFrame(d)

所以作为输出我得到：

    data_1  data_2  data_3
a   1.0      5.0    9.0
b   2.0      6.0    10.0
c   3.0      7.0    11.0
cls 1.0      0.0    1.0

我想做的是根据“ cls”行中报告的值创建一个过滤器，然后根据两个可能的类将数据帧分为两个

    data_1  data_3
a   1.0      9.0
b   2.0      10.0
c   3.0      11.0
cls 1.0      1.0

和

        data_2
a        5.0
b        6.0
c        7.0
cls      0.0

最佳答案

将boolean indexing与loc一起使用-首先按条件选择所有列，然后按index选择行：

df1 = df.loc[:, df.loc['class'] == 1]
df2 = df.loc[:, df.loc['class'] == 0]

还可以通过行class的所有唯一值来创建字典：

s = df.loc['class']
dfs = {k:df[v.index] for k, v in s.groupby(s)}
print (dfs)

{0.0:        data_2
a         5.0
b         6.0
c         7.0
d         8.0
class     0.0, 1.0:        data_1  data_3
a         1.0     9.0
b         2.0    10.0
c         3.0    11.0
d         4.0    12.0
class     1.0     1.0}

print (list(dfs.keys()))
[0.0, 1.0]

print (dfs[0])
       data_2
a         5.0
b         6.0
c         7.0
d         8.0
class     0.0:

print (dfs[1])
       data_1  data_3
a         1.0     9.0
b         2.0    10.0
c         3.0    11.0
d         4.0    12.0
class     1.0     1.0

关于python - pandas:基于行值的列样本，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/48503346/