[每日一练]每位教师所教授科目的种类与agg聚合参数的介绍

该题目来源于力扣：

2356. 每位教师所教授的科目种类的数量 - 力扣（LeetCode）

题目要求：

表: Teacher

+-------------+------+
| Column Name | Type |
+-------------+------+
| teacher_id  | int  |
| subject_id  | int  |
| dept_id     | int  |
+-------------+------+
在 SQL 中，(subject_id, dept_id) 是该表的主键。
该表中的每一行都表示带有 teacher_id 的教师在系 dept_id 中教授科目 subject_id。
 

查询每位老师在大学里教授的科目种类的数量。

以 任意顺序 返回结果表。

查询结果格式示例如下。

 例 1:

输入: 
Teacher 表:
+------------+------------+---------+
| teacher_id | subject_id | dept_id |
+------------+------------+---------+
| 1          | 2          | 3       |
| 1          | 2          | 4       |
| 1          | 3          | 3       |
| 2          | 1          | 1       |
| 2          | 2          | 1       |
| 2          | 3          | 1       |
| 2          | 4          | 1       |
+------------+------------+---------+
输出:  
+------------+-----+
| teacher_id | cnt |
+------------+-----+
| 1          | 2   |
| 2          | 4   |
+------------+-----+
解释: 
教师 1:
  - 他在 3、4 系教科目 2。
  - 他在 3 系教科目 3。
教师 2:
  - 他在 1 系教科目 1。
  - 他在 1 系教科目 2。
  - 他在 1 系教科目 3。
  - 他在 1 系教科目 4。

思路流程：

这道题很简单，经典的分组聚合问题

• 首先要按照教师id进行分组
• 然后利用agg函数进行聚合计算
• 由于分组函数会将分组的列作为索引，所以我们需要利用reset_index()函数进行索引重置

关于agg聚合操作的参数介绍：

• 'sum'：计算指定列的总和。
• 'mean'：计算指定列的平均值。
• 'median'：计算指定列的中位数。
• 'min'：找出指定列的最小值。
• 'max'：找出指定列的最大值。
• 'count'：计算指定列的非缺失值数量。
• 'nunique'：计算指定列的唯一值数量。
• 'std'：计算指定列的标准差。
• 'var'：计算指定列的方差。
• 'first'：返回指定列的第一个非缺失值。
• 'last'：返回指定列的最后一个非缺失值。

由此我们可以看出，利用"nunique"参数符合题目聚合目标

代码实现：

import pandas as pd

def count_unique_subjects(teacher: pd.DataFrame) -> pd.DataFrame:
    data=teacher.groupby('teacher_id').agg({'subject_id':'nunique'}).reset_index().rename(columns={'subject_id':'cnt'})
    return data[['teacher_id','cnt']]