diff --git a/lab-sql-python-connection.ipynb b/lab-sql-python-connection.ipynb new file mode 100644 index 0000000..0b47140 --- /dev/null +++ b/lab-sql-python-connection.ipynb @@ -0,0 +1,761 @@ +{ + "cells": [ + { + "cell_type": "code", + "execution_count": 1, + "id": "3b67f7d1", + "metadata": {}, + "outputs": [], + "source": [ + "import pymysql\n", + "import pandas as pd\n", + "import getpass\n", + "\n", + "password = getpass.getpass(\"MySQL password: \")\n", + "\n", + "conn = pymysql.connect(\n", + " host=\"127.0.0.1\",\n", + " user=\"root\",\n", + " password=password,\n", + " database=\"sakila\"\n", + ")" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "id": "c22cb79f", + "metadata": {}, + "outputs": [], + "source": [ + "def rentals_month(conn, month, year):\n", + " query = \"\"\"\n", + " SELECT\n", + " rental_id,\n", + " rental_date,\n", + " inventory_id,\n", + " customer_id,\n", + " return_date,\n", + " staff_id,\n", + " last_update\n", + " FROM rental\n", + " WHERE YEAR(rental_date) = %s\n", + " AND MONTH(rental_date) = %s\n", + " \"\"\"\n", + " \n", + " cursor = conn.cursor()\n", + " cursor.execute(query, (year, month))\n", + " rows = cursor.fetchall()\n", + " \n", + " columns = [\n", + " \"rental_id\",\n", + " \"rental_date\",\n", + " \"inventory_id\",\n", + " \"customer_id\",\n", + " \"return_date\",\n", + " \"staff_id\",\n", + " \"last_update\"\n", + " ]\n", + " \n", + " return pd.DataFrame(rows, columns=columns)\n" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "id": "4e9ea76a", + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
rental_idrental_dateinventory_idcustomer_idreturn_datestaff_idlast_update
012005-05-24 22:53:303671302005-05-26 22:04:3012006-02-15 21:30:53
122005-05-24 22:54:3315254592005-05-28 19:40:3312006-02-15 21:30:53
232005-05-24 23:03:3917114082005-06-01 22:12:3912006-02-15 21:30:53
342005-05-24 23:04:4124523332005-06-03 01:43:4122006-02-15 21:30:53
452005-05-24 23:05:2120792222005-06-02 04:33:2112006-02-15 21:30:53
\n", + "
" + ], + "text/plain": [ + " rental_id rental_date inventory_id customer_id \\\n", + "0 1 2005-05-24 22:53:30 367 130 \n", + "1 2 2005-05-24 22:54:33 1525 459 \n", + "2 3 2005-05-24 23:03:39 1711 408 \n", + "3 4 2005-05-24 23:04:41 2452 333 \n", + "4 5 2005-05-24 23:05:21 2079 222 \n", + "\n", + " return_date staff_id last_update \n", + "0 2005-05-26 22:04:30 1 2006-02-15 21:30:53 \n", + "1 2005-05-28 19:40:33 1 2006-02-15 21:30:53 \n", + "2 2005-06-01 22:12:39 1 2006-02-15 21:30:53 \n", + "3 2005-06-03 01:43:41 2 2006-02-15 21:30:53 \n", + "4 2005-06-02 04:33:21 1 2006-02-15 21:30:53 " + ] + }, + "execution_count": 3, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "may_rentals = rentals_month(conn, 5, 2005)\n", + "may_rentals.head()\n" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "id": "4da7058f", + "metadata": {}, + "outputs": [], + "source": [ + "def rental_count_month(rentals_df, month, year):\n", + " col = f\"rentals_{month:02d}_{year}\"\n", + " return (\n", + " rentals_df\n", + " .groupby(\"customer_id\")\n", + " .size()\n", + " .reset_index(name=col)\n", + " )\n" + ] + }, + { + "cell_type": "code", + "execution_count": 10, + "id": "6325fccd", + "metadata": {}, + "outputs": [], + "source": [ + "def compare_rentals(df_a, df_b):\n", + " col_a = [c for c in df_a.columns if c.startswith(\"rentals_\")][0]\n", + " col_b = [c for c in df_b.columns if c.startswith(\"rentals_\")][0]\n", + " \n", + " result = df_a.merge(df_b, on=\"customer_id\", how=\"inner\")\n", + " result[\"difference\"] = result[col_b] - result[col_a]\n", + " return result\n" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "id": "f7416db1", + "metadata": {}, + "outputs": [], + "source": [ + "may_rentals = rentals_month(conn, 5, 2005)\n", + "jun_rentals = rentals_month(conn, 6, 2005)\n" + ] + }, + { + "cell_type": "code", + "execution_count": 7, + "id": "f0e6b870", + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
customer_idrentals_05_2005
012
121
232
353
463
\n", + "
" + ], + "text/plain": [ + " customer_id rentals_05_2005\n", + "0 1 2\n", + "1 2 1\n", + "2 3 2\n", + "3 5 3\n", + "4 6 3" + ] + }, + "execution_count": 7, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "may_counts = rental_count_month(may_rentals, 5, 2005)\n", + "jun_counts = rental_count_month(jun_rentals, 6, 2005)\n", + "\n", + "may_counts.head()\n" + ] + }, + { + "cell_type": "code", + "execution_count": 8, + "id": "bf0ec5d7", + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
customer_idrentals_06_2005
017
121
234
346
455
\n", + "
" + ], + "text/plain": [ + " customer_id rentals_06_2005\n", + "0 1 7\n", + "1 2 1\n", + "2 3 4\n", + "3 4 6\n", + "4 5 5" + ] + }, + "execution_count": 8, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "jun_counts.head()" + ] + }, + { + "cell_type": "code", + "execution_count": 11, + "id": "59e15e27", + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
customer_idrentals_05_2005rentals_06_2005difference
01275
12110
23242
35352
46341
\n", + "
" + ], + "text/plain": [ + " customer_id rentals_05_2005 rentals_06_2005 difference\n", + "0 1 2 7 5\n", + "1 2 1 1 0\n", + "2 3 2 4 2\n", + "3 5 3 5 2\n", + "4 6 3 4 1" + ] + }, + "execution_count": 11, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "comparison = compare_rentals(may_counts, jun_counts)\n", + "comparison.head()" + ] + }, + { + "cell_type": "code", + "execution_count": 12, + "id": "9bb2318d", + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
customer_idrentals_05_2005rentals_06_2005difference
3864541109
389457198
178213198
248295198
2327187
194234187
218260187
322380187
481561297
345406176
\n", + "
" + ], + "text/plain": [ + " customer_id rentals_05_2005 rentals_06_2005 difference\n", + "386 454 1 10 9\n", + "389 457 1 9 8\n", + "178 213 1 9 8\n", + "248 295 1 9 8\n", + "23 27 1 8 7\n", + "194 234 1 8 7\n", + "218 260 1 8 7\n", + "322 380 1 8 7\n", + "481 561 2 9 7\n", + "345 406 1 7 6" + ] + }, + "execution_count": 12, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "comparison.sort_values(\"difference\", ascending=False).head(10)" + ] + }, + { + "cell_type": "code", + "execution_count": 13, + "id": "28fa7c4c", + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
customer_idrentals_05_2005rentals_06_2005difference
17320761-5
23027462-4
21025051-4
16719851-4
50959662-4
111451-4
13616162-4
39146041-3
18622252-3
47455152-3
\n", + "
" + ], + "text/plain": [ + " customer_id rentals_05_2005 rentals_06_2005 difference\n", + "173 207 6 1 -5\n", + "230 274 6 2 -4\n", + "210 250 5 1 -4\n", + "167 198 5 1 -4\n", + "509 596 6 2 -4\n", + "11 14 5 1 -4\n", + "136 161 6 2 -4\n", + "391 460 4 1 -3\n", + "186 222 5 2 -3\n", + "474 551 5 2 -3" + ] + }, + "execution_count": 13, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "comparison.sort_values(\"difference\").head(10)" + ] + } + ], + "metadata": { + "kernelspec": { + "display_name": "base", + "language": "python", + "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.13.9" + } + }, + "nbformat": 4, + "nbformat_minor": 5 +}