Update 596. Classes More Than 5 Students.py

Author: iamAntimPal

LeetCode SQL 50 Solution/596. Classes More Than 5 Students/596. Classes More Than 5 Students.py

@@ -0,0 +1,7 @@
+import pandas as pd
+
+def classes_with_five_or_more_students(courses: pd.DataFrame) -> pd.DataFrame:
+    # Group by 'class' and count the number of students
+    result = courses.groupby('class').filter(lambda x: len(x) >= 5)
+    # Return only the distinct class names
+    return result[['class']].drop_duplicates().reset_index(drop=True)