浏览代码

Adding solution for 2882

Vinicius Teshima 2 周之前
父节点
当前提交
674f8215bf
共有 1 个文件被更改,包括 52 次插入0 次删除
  1. 52 0
      Python/2882.py

+ 52 - 0
Python/2882.py

@@ -0,0 +1,52 @@
+# 2882. Drop Duplicate Rows
+
+import sys
+
+import pandas as pd
+
+def dropDuplicateEmails(customers: pd.DataFrame) -> pd.DataFrame:
+    return customers.drop_duplicates(subset=['email'], keep='first')
+
+
+def main() -> int:
+
+    def r(customers: pd.DataFrame, exp: pd.DataFrame) -> None:
+        res: pd.DataFrame = dropDuplicateEmails(customers)
+        print(f"dropDuplicateEmails({customers}) = {res} | exp: {exp}")
+        pass
+
+    r(
+        pd.DataFrame.from_dict(
+            {
+                "customer_id": [1, 2, 3, 4, 5, 6],
+                "name": ["Ella", "David", "Zachary", "Alice", "Finn", "Violet"],
+                "email": [
+                    "emily@example.com",
+                    "michael@example.com",
+                    "sarah@example.com",
+                    "john@example.com",
+                    "john@example.com",
+                    "alice@example.com",
+                ],
+            }
+        ),
+        pd.DataFrame.from_dict(
+            {
+                "customer_id": [1, 2, 3, 4, 6],
+                "name": ["Ella", "David", "Zachary", "Alice", "Violet"],
+                "email": [
+                    "emily@example.com",
+                    "michael@example.com",
+                    "sarah@example.com",
+                    "john@example.com",
+                    "alice@example.com",
+                ],
+            }
+        ),
+    )
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())