python - 如何按行值过滤和分组条目

Question

我有以下数据框:

df = 
ID   GROUP_1  GROUP_2  GROUP_3  GRADE
1A   AAA      BBB      AAA      5
1B   BBB      BBB      CCC      4
1C   AAA      BBB      BBB      4

我想计算成绩的行数 5 , 4 , .., 1对于 GROUP 的每个唯一值。请注意，在第 1 行中，AAA 出现了 2 次。 ，不过我数了一次。

给定数据集的预期输出如下:

GROUP  GRADE_1   GRADE_2   GRADE_3   GRADE_4   GRADE_5
AAA    0         0         0         1         1
BBB    0         0         0         2         1
CCC    0         0         0         1         0

我有下面给出的代码，它工作正常，不包括分组( groupby('GRADE') )。我不知道如何按 GRADE 对结果进行分组并创建列GRADE_1 , GRADE_2 ,..., GRADE_5 .

df.groupby('GRADE').filter(regex="^GROUP").stack().reset_index(level=1, drop=True).reset_index().drop_duplicates()[0].value_counts()

Answer 1

试试这个:

In [56]: df
Out[56]:
   ID GROUP_1 GROUP_2 GROUP_3  GRADE
0  1A     AAA     BBB     AAA      5
1  1B     BBB     BBB     CCC      4
2  1C     AAA     BBB     BBB      4

In [57]: (df.set_index('GRADE')
    ...:    .filter(like='GROUP_')
    ...:    .stack()
    ...:    .to_frame('GROUP')
    ...:    .reset_index()
    ...:    .pivot_table(index='GROUP', columns='GRADE', aggfunc='size', fill_value=0)
    ...: )
    ...:
Out[57]:
GRADE  4  5
GROUP
AAA    1  2
BBB    4  1
CCC    1  0