python - 基于逻辑条件的 Pandas DataFrame 切片？

Question

我有这个名为数据的数据框:

       Subjects  Professor  StudentID
8     Chemistry       Jane        999
1     Chemistry       Jane       3455
0     Chemistry     Joseph       1234
2       History       Jane       3455
6       History      Smith        323
7       History      Smith        999
3   Mathematics        Doe      56767
10  Mathematics   Einstein       3455
5       Physics   Einstein       2834
9       Physics      Smith        323
4       Physics      Smith        999

我想运行此查询“Professors with at least 2 classes with 2 or more same students”。期望的输出

Smith: Physics, History, 323, 999

我熟悉 SQL 并且可以轻松完成此操作，但我仍然是 Python 的初学者。如何在 Python 中实现此输出？另一种思路是将此数据框转换为 SQL 数据库，并通过 python 具有 SQL 接口(interface)来运行查询。有办法实现吗？

Answer 1

students_and_subjects = df.groupby(
                               ['Professor', 'Subjects']
                           ).StudentID.nunique().ge(2) \
                          .groupby(level='Professor').sum().ge(2)

df[df.Professor.map(students_and_subjects)]