LeetCode Pyspark Questions Solution

Author: bitoollearner

Solved/1212. Team Scores in Football Tournament (Medium)-(Solved).ipynb

@@ -4,7 +4,10 @@
    "cell_type": "markdown",
    "metadata": {
     "application/vnd.databricks.v1+cell": {
-     "cellMetadata": {},
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
      "inputWidgets": {},
      "nuid": "82383caa-9b15-409a-9acb-968b851c477b",
      "showTitle": false,
@@ -21,7 +24,10 @@
    "execution_count": 0,
    "metadata": {
     "application/vnd.databricks.v1+cell": {
-     "cellMetadata": {},
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
      "inputWidgets": {},
      "nuid": "83322872-9913-4063-98cf-4aaea5fd3337",
      "showTitle": false,
@@ -40,7 +46,10 @@
    "cell_type": "markdown",
    "metadata": {
     "application/vnd.databricks.v1+cell": {
-     "cellMetadata": {},
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
      "inputWidgets": {},
      "nuid": "e475f002-2b95-4d27-958c-aeab86c066ed",
      "showTitle": false,
@@ -124,7 +133,10 @@
    "execution_count": 0,
    "metadata": {
     "application/vnd.databricks.v1+cell": {
-     "cellMetadata": {},
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
      "inputWidgets": {},
      "nuid": "a073d765-b3a4-4bda-a02f-ac5e3671a222",
      "showTitle": false,
@@ -158,25 +170,154 @@
     "matches_df_1212 = spark.createDataFrame(matches_data_1212, matches_columns_1212)\n",
     "matches_df_1212.show()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 0,
+   "metadata": {
+    "application/vnd.databricks.v1+cell": {
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
+     "inputWidgets": {},
+     "nuid": "0d53ec9e-0db8-4070-90c2-0cffb4ef13f5",
+     "showTitle": false,
+     "tableResultSettingsMap": {},
+     "title": ""
+    }
+   },
+   "outputs": [],
+   "source": [
+    "host_df_1212 = matches_df_1212.select(\n",
+    "    col(\"host_team\").alias(\"team_id\"),\n",
+    "    when(col(\"host_goals\") > col(\"guest_goals\"), 3)\n",
+    "    .when(col(\"host_goals\") == col(\"guest_goals\"), 1)\n",
+    "    .otherwise(0)\n",
+    "    .alias(\"points\")\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 0,
+   "metadata": {
+    "application/vnd.databricks.v1+cell": {
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
+     "inputWidgets": {},
+     "nuid": "8a77b973-b826-4986-9634-7fc0ec8cfe48",
+     "showTitle": false,
+     "tableResultSettingsMap": {},
+     "title": ""
+    }
+   },
+   "outputs": [],
+   "source": [
+    "guest_df_1212 = matches_df_1212.select(\n",
+    "    col(\"guest_team\").alias(\"team_id\"),\n",
+    "    when(col(\"guest_goals\") > col(\"host_goals\"), 3)\n",
+    "    .when(col(\"guest_goals\") == col(\"host_goals\"), 1)\n",
+    "    .otherwise(0)\n",
+    "    .alias(\"points\")\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 0,
+   "metadata": {
+    "application/vnd.databricks.v1+cell": {
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
+     "inputWidgets": {},
+     "nuid": "3efb888d-7a40-4caa-9389-651afd14488b",
+     "showTitle": false,
+     "tableResultSettingsMap": {},
+     "title": ""
+    }
+   },
+   "outputs": [],
+   "source": [
+    "points_df_1212 = host_df_1212.union(guest_df_1212)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 0,
+   "metadata": {
+    "application/vnd.databricks.v1+cell": {
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
+     "inputWidgets": {},
+     "nuid": "5056a058-2a62-46e6-95e1-f388b158b69b",
+     "showTitle": false,
+     "tableResultSettingsMap": {},
+     "title": ""
+    }
+   },
+   "outputs": [],
+   "source": [
+    "team_points_df_1212 = points_df_1212.groupBy(\"team_id\").agg(sum(\"points\").alias(\"num_points\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 0,
+   "metadata": {
+    "application/vnd.databricks.v1+cell": {
+     "cellMetadata": {
+      "byteLimit": 2048000,
+      "rowLimit": 10000
+     },
+     "inputWidgets": {},
+     "nuid": "326f1ae2-4987-49b0-9e98-976e79b121f1",
+     "showTitle": false,
+     "tableResultSettingsMap": {},
+     "title": ""
+    }
+   },
+   "outputs": [],
+   "source": [
+    "team_points_df_1212\\\n",
+    "    .join(teams_df_1212, on=\"team_id\", how=\"right\").fillna(0)\\\n",
+    "        .select(\"team_id\", \"team_name\", \"num_points\") \\\n",
+    "            .orderBy(col(\"num_points\").desc(), col(\"team_id\")).show()"
+   ]
   }
  ],
  "metadata": {
   "application/vnd.databricks.v1+notebook": {
-   "computePreferences": null,
+   "computePreferences": {
+    "hardware": {
+     "accelerator": null,
+     "gpuPoolId": null,
+     "memory": null
+    }
+   },
    "dashboards": [],
    "environmentMetadata": {
     "base_environment": "",
-    "environment_version": "1"
+    "environment_version": "2"
    },
    "inputWidgetPreferences": null,
    "language": "python",
    "notebookMetadata": {
     "pythonIndentUnit": 4
    },
-   "notebookName": "1212. Team Scores in Football Tournament (Medium)",
+   "notebookName": "1212. Team Scores in Football Tournament (Medium)-(Solved)",
    "widgets": {}
+  },
+  "language_info": {
+   "name": "python"
   }
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
+}