Pandas groupby 并获取数据框中多列的唯一性

Question

我有一个如下所示的数据框

stu_id,Mat_grade,sci_grade,eng_grade
1,A,C,A
1,A,C,A
1,B,C,A
1,C,C,A
2,D,B,B
2,D,C,B
2,D,D,C
2,D,A,C

tf = pd.read_clipboard(sep=',')

我的objective是

a) 找出学生在 Mat_grade、sci_grade 和 eng_grade

下获得了多少个不同的唯一成绩

所以，我尝试了下面的方法

tf['mat_cnt'] = tf.groupby(['stu_id'])['Mat_grade'].nunique()
tf['sci_cnt'] = tf.groupby(['stu_id'])['sci_grade'].nunique()
tf['eng_cnt'] = tf.groupby(['stu_id'])['eng_grade'].nunique()

但这并没有提供预期的输出。因为，我有超过 100K 个唯一 ID，任何高效和优雅的解决方案都非常有帮助

我希望我的输出如下所示

Answer 1

您可以在列表中指定列名称，并为列 cols 调用 DataFrameGroupBy.nunique 和 rename:

cols = ['Mat_grade','sci_grade', 'eng_grade']
new = ['mat_cnt','sci_cnt','eng_cnt']
d = dict(zip(cols, new))
df = tf.groupby(['stu_id'], as_index=False)[cols].nunique().rename(columns=d)
print (df)
   stu_id  mat_cnt  sci_cnt  eng_cnt
0       1        3        1        1
1       2        1        4        2

另一个想法是使用命名聚合：

cols = ['Mat_grade','sci_grade', 'eng_grade']
new = ['mat_cnt','sci_cnt','eng_cnt']
d = {v: (k,'nunique') for k, v in zip(cols, new)}
print (d)
{'mat_cnt': ('Mat_grade', 'nunique'), 
 'sci_cnt': ('sci_grade', 'nunique'), 
 'eng_cnt': ('eng_grade', 'nunique')}

df = tf.groupby(['stu_id'], as_index=False).agg(**d)
print (df)
   stu_id  mat_cnt  sci_cnt  eng_cnt
0       1        3        1        1
1       2        1        4        2

Pandas groupby 并获取数据框中多列的唯一性

Pandas groupby and get nunique of multiple columns in a dataframe

python

list

dataframe

pandas

pandas-groupby