Skip to content

Commit

Permalink
Reuse arg_parse code in demo notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
@mccalluc
mccalluc committed Oct 21, 2024
1 parent 3d97978 commit 4fdf1a5
Showing 1 changed file with 8 additions and 206 deletions.
214 changes: 8 additions & 206 deletions demo.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,227 +4,29 @@
"cell_type": "markdown",
"metadata": {},
"source": [
"This is just to demonstrate how the analysis will work for the POC, and also to show what the final exported notebook will look like. When that is implemented, get rid of this!\n",
"\n",
"First, generate a fake dataset. In the future, let's check it in and use it if the [`--demo` flag](https://github.com/opendp/dp-creator-ii/issues/7) is given."
"This is just to demonstrate how the analysis will work for the POC, and also to show what the final exported notebook will look like. When that is implemented, get rid of this!"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Make mock \n",
"# Make fake dataset \n",
"\n",
"When [Add `--demo` CLI option](https://github.com/opendp/dp-creator-ii/pull/61) is merged, reference that code and delete these cells.\n"
"This will not be part of the final notebook!\n"
]
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import csv\n",
"import random\n",
"\n",
"random.seed(0) # So the mock data will be stable across runs.\n",
"\n",
"def clip(n, lower, upper):\n",
" return max(min(n, upper), lower)\n",
"\n",
"csv_path = '/tmp/demo.csv'\n",
"\n",
"with open(csv_path, 'w', newline='') as demo_handle:\n",
" fields = ['student_id', 'class_year', 'hw_number', 'grade']\n",
" writer = csv.DictWriter(demo_handle, fieldnames=fields)\n",
" writer.writeheader()\n",
" for student_id in range(1, 100):\n",
" class_year = int(clip(random.gauss(2, 1), 1, 4))\n",
" mean_grade = random.gauss(80, 5) + class_year * 2\n",
" for hw_number in range(1, 10):\n",
" grade = int(clip(random.gauss(mean_grade, 5), 0, 100))\n",
" writer.writerow({\n",
" 'student_id': student_id,\n",
" 'class_year': class_year,\n",
" 'hw_number': hw_number,\n",
" 'grade': grade,\n",
" })"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Sanity check:"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div><style>\n",
".dataframe > thead > tr,\n",
".dataframe > tbody > tr {\n",
" text-align: right;\n",
" white-space: pre-wrap;\n",
"}\n",
"</style>\n",
"<small>shape: (891, 4)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>student_id</th><th>class_year</th><th>hw_number</th><th>grade</th></tr><tr><td>i64</td><td>i64</td><td>i64</td><td>i64</td></tr></thead><tbody><tr><td>1</td><td>2</td><td>1</td><td>73</td></tr><tr><td>1</td><td>2</td><td>2</td><td>78</td></tr><tr><td>1</td><td>2</td><td>3</td><td>71</td></tr><tr><td>1</td><td>2</td><td>4</td><td>76</td></tr><tr><td>1</td><td>2</td><td>5</td><td>77</td></tr><tr><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td></tr><tr><td>99</td><td>2</td><td>5</td><td>78</td></tr><tr><td>99</td><td>2</td><td>6</td><td>74</td></tr><tr><td>99</td><td>2</td><td>7</td><td>75</td></tr><tr><td>99</td><td>2</td><td>8</td><td>72</td></tr><tr><td>99</td><td>2</td><td>9</td><td>82</td></tr></tbody></table></div>"
],
"text/plain": [
"shape: (891, 4)\n",
"┌────────────┬────────────┬───────────┬───────┐\n",
"│ student_id ┆ class_year ┆ hw_number ┆ grade │\n",
"│ --- ┆ --- ┆ --- ┆ --- │\n",
"│ i64 ┆ i64 ┆ i64 ┆ i64 │\n",
"╞════════════╪════════════╪═══════════╪═══════╡\n",
"│ 1 ┆ 2 ┆ 1 ┆ 73 │\n",
"│ 1 ┆ 2 ┆ 2 ┆ 78 │\n",
"│ 1 ┆ 2 ┆ 3 ┆ 71 │\n",
"│ 1 ┆ 2 ┆ 4 ┆ 76 │\n",
"│ 1 ┆ 2 ┆ 5 ┆ 77 │\n",
"│ … ┆ … ┆ … ┆ … │\n",
"│ 99 ┆ 2 ┆ 5 ┆ 78 │\n",
"│ 99 ┆ 2 ┆ 6 ┆ 74 │\n",
"│ 99 ┆ 2 ┆ 7 ┆ 75 │\n",
"│ 99 ┆ 2 ┆ 8 ┆ 72 │\n",
"│ 99 ┆ 2 ┆ 9 ┆ 82 │\n",
"└────────────┴────────────┴───────────┴───────┘"
]
},
"execution_count": 2,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"import polars\n",
"from dp_creator_ii.argparse_helpers import _get_demo_csv_contrib\n",
"\n",
"lf = polars.scan_csv(csv_path)\n",
"lf.collect()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"TODO: Plot function from in https://github.com/opendp/dp-creator-ii/pull/35... but see farther down in the notebook: This can probably be further simplified."
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [],
"source": [
"import matplotlib.pyplot as plt\n",
"import numpy as np\n",
"\n",
"\n",
"def plot_error_bars_with_cutoff(\n",
" y_values, x_min_label=\"min\", x_max_label=\"max\", y_cutoff=0, y_error=0\n",
"):\n",
" x_values = 0.5 + np.arange(len(y_values))\n",
" x_values_above = []\n",
" x_values_below = []\n",
" y_values_above = []\n",
" y_values_below = []\n",
" for x, y in zip(x_values, y_values):\n",
" if y < y_cutoff:\n",
" x_values_below.append(x)\n",
" y_values_below.append(y)\n",
" else:\n",
" x_values_above.append(x)\n",
" y_values_above.append(y)\n",
"\n",
" figure, axes = plt.subplots()\n",
" color = \"skyblue\"\n",
" shared = {\n",
" \"width\": 0.8,\n",
" \"edgecolor\": color,\n",
" \"linewidth\": 1,\n",
" \"yerr\": y_error,\n",
" }\n",
" axes.bar(x_values_above, y_values_above, color=color, **shared)\n",
" axes.bar(x_values_below, y_values_below, color=\"white\", **shared)\n",
" axes.hlines([y_cutoff], 0, len(y_values), colors=[\"black\"], linestyles=[\"dotted\"])\n",
"\n",
" axes.set(xlim=(0, len(y_values)), ylim=(0, max(y_values)))\n",
" axes.get_xaxis().set_ticks(\n",
" ticks=[x_values[0], x_values[-1]],\n",
" labels=[x_min_label, x_max_label],\n",
" )\n",
" axes.get_yaxis().set_ticks([])"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div><style>\n",
".dataframe > thead > tr,\n",
".dataframe > tbody > tr {\n",
" text-align: right;\n",
" white-space: pre-wrap;\n",
"}\n",
"</style>\n",
"<small>shape: (4, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>class_year</th><th>len</th></tr><tr><td>i64</td><td>u32</td></tr></thead><tbody><tr><td>1</td><td>414</td></tr><tr><td>2</td><td>297</td></tr><tr><td>3</td><td>126</td></tr><tr><td>4</td><td>54</td></tr></tbody></table></div>"
],
"text/plain": [
"shape: (4, 2)\n",
"┌────────────┬─────┐\n",
"│ class_year ┆ len │\n",
"│ --- ┆ --- │\n",
"│ i64 ┆ u32 │\n",
"╞════════════╪═════╡\n",
"│ 1 ┆ 414 │\n",
"│ 2 ┆ 297 │\n",
"│ 3 ┆ 126 │\n",
"│ 4 ┆ 54 │\n",
"└────────────┴─────┘"
]
},
"execution_count": 4,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"class_years = lf.group_by('class_year').len().collect().sort('class_year')\n",
"class_years"
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [
{
"data": {
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAhAAAAGdCAYAAABDxkoSAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjkuMiwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy8hTgPZAAAACXBIWXMAAA9hAAAPYQGoP6dpAAAWvklEQVR4nO3df4xd5Xng8efOjMceOJ7BiMQwwtZQhJpYZdkkgEWmItJihYSGLhslKZRSlqRYFQtJSlIlSIXLdlESUJTSgBXibhIS1NI0rExa0gZRQHiVgGENIVtgg4ncxK47jsD2jMcz9sz9sX+4Pp5rD8WPmWF+fT5SpJzXr+99Z87JnG/OneNTaTabzQAASGib6QUAAHOPgAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBI68hMbjQasWPHjli6dGlUKpXpWhMAMIWazWbs3bs3ent7o61taq4dpAJix44dsWLFiil5YwDgrbVt27Y4/fTTp+S1UgGxdOnScgHd3d1TsgAAFpadI7X4yy2DR41feVZPLD8hdVriGA0NDcWKFSvK8/hUSO2pQx9bdHd3CwgAjstIRy2WFI2jxpd2d0e3gJhWU/nrB36JEgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEjzT34BMOcNjtVjtNYst7s6KtHT2T6DK5r/BAQAc9rgWD3Wv7g76of7IdorEWtXLRMR08hHGADMaaO1Zks8RETUm9FyRYKpJyAAgDQBAQCkCQgAIE1AAABpAgIASBMQAECagAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQ1jHTCwBgfhscq8dorVluD4/XZ3A1rY5cW1dHJXo622dwRXOHgABg2gyO1WP9i7ujfvgcPWsufU+2tvZKxNpVy0TEMZgt+xGAeWi01mw5QUdENGZmKUeZbG31ZrRckeD1CQgAIE1AAABpAgIASBMQAECagAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGmehQHArOVhV7OXgABgVvKwq9nNRxgAzEoedjW7CQgAIE1AAABpAgIASBMQAECagAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkdcz0AgCYmwbH6jFaa5bbXR2V6Olsn/b3HR5vxMBIbcJ2fdrfk6MJCADSBsfqsf7F3VE/3A/RXolYu2rZtEfEhq1DLe/rUvrM8H0HIG201mw5iUdE1JvRckViuhz5vo1pf0cmIyAAgDQBAQCkCQgAIE1AAABp7sIAYFZwe+bcIiAAmBXcnjm32D8AzApuz5xbBAQAkCYgAIA0AQEApAkIACDNXRgATBm3Yi4cAgKAKeNWzIXDvgVgyrgVc+EQEABAmoAAANIEBACQ5pcoATjK4Fg9RmsHf6Ghq6MSPZ3tM7wiZhsBAUCLwbF6rH9xd/kLke2ViLWrlokIWvgIA4AWo7Vmy90U9WaUVyPgEAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaR6mBbCATHzKZsT8ftLm8HgjBkZq5fZ8/lpngoAAWCCOfMpmxPx+0uaGrUML5mudCT7CAFggjnzKZsT8ftLmQvpaZ4KAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQ5lkYACwYEx+wNTxen+HVzG0CAoAFY+IDtlyCf3N8/wBYMCY+YKsxc8uYFwQEAJAmIACANAEBAKQJCAAgzV0YALyhibc/Htyev7dAHvm1dnVUoqezfQZXNDsJCADe0MTbHyPm9+XrI7/W9krE2lXLRMQR5vMxAMAUmXhCjZjft0Ae+bXWmxGjtebkkxcwAQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkCQgAIE1AAABpAgIASBMQAECagAAA0gQEAJAmIACANAEBAKR1zPQCAJhZw+ONGBipTdiuz+BqmCsEBMACt2HrUNSbh7ddmuZYOE4AFriJ8RAR0ZiZZTDHCAgAIE1AAABpAgIASBMQAECagAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkdcz0AgCYHoNj9RitNcvt4fH6DK6G+UZAAMxDg2P1WP/i7qgf7geXnJlSjieAeWi01myJh4iIxswshXlKQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkCQgAIE1AAABpAgIASBMQAECagAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkCQgAIE1AAABpAgIASBMQAECagAAA0gQEAJDWMdMLAIDZbni8EQMjtXK7q6MSPZ3tM7iimScgAOANbNg6FPXm4e32SsTaVcsWdET4CAMA3sDEeDi0PVprTj55gRAQAECagAAA0gQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkCQgAIE1AAABpAgIASOuY6QUAkDM4Vo/RWrPc7uqoRE9n+wyuaGEaHm/EwEit3O7qqERELJh9IyAA5pDBsXqsf3F31A+fo6K9ErF21bJ5e6KarTZsHWrZD20RUanEgtk3PsIAmENGa82WE1TEwRPWxP/Xy1vjyP3QmGRsPu8bAQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkCQgAIE1AAABpHqYFMA8c+WTI4fH6DK6GhUBAAMwDkz0ZEqaTYwxgHpjsyZAwnQQEAJAmIACANAEBAKQJCAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQBAQCkCQgAIE1AAABpAgIASBMQAECagAAA0jpmegEAvL7BsXqM1prl9vB4fQZXw/EYHm/EwEit3O7qqEREtOzXro5K9HS2v+VrezMEBMAsNThWj/Uv7o764fOMy8Zz0IatQ0ftw0olWsbaKxFrVy2bUxHhWASYpUZrzZaTTEREY2aWwpsw2T48cqzebL0iMRcICAAgTUAAAGkCAgBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaZ6FATBLeHAWc4mAAJgFPDiLucbxCTALeHAWc42AAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDQP0wKAWWB4vBEDI7Vyu6ujEhHR8oTWro5K9HS2v+Vrm4yAAIBZYMPWoaOexlqpRMtYeyVi7aplsyIifIQBALPAZE9jPXKs3my9IjGTBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQJiAAgDTPwgCYAYNj9ZZ/knh4vD6Dq4E8AQHwFhscq8f6F3cf9eAkmEscswBvsdFac9IHJ8FcIiAAgDQBAQCkCQgAIE1AAABp7sIAmGZu2WS6HXmMdXVUoqezfVrfU0AATCO3bDLdJjvG2isRa1ctm9aIcBwDTCO3bDLdJjvG6s1ouSIxHQQEAJAmIACANAEBAKT5JUoAmEOGxxsxMFKbsD0zd/UICACYQzZsHZoVd/X4CAMA5pDZclePgAAA0gQEAJAmIACANAEBAKS5CwNgCnlwFguFgACYIh6cxULi2AaYIh6cxUIiIACANAEBAKQJCAAgTUAAAGnuwgBmtSNvi+zqqERPZ/sxzYuIY/q7QJ6AAGatyW6LbK9ErF21rCUEXu/2yUol3vDvAsfHRxjArDXZbZH1ZutVhdeb14ijn1o42d8Fjo+AAADSBAQAkCYgAIA0AQEApLkLA1hQhscbMTBSK7czt4W+0TxP3mQ2eW3/4eN8fGzqj00BASwoG7YOHddtoccyzyVdZpO/+8Vw+d/H9+2d8td3vAMLyvHeFnos8zx5k9nqyON5KggIACBNQAAAacf1OxD79u2LpUuXRqVy8N+aHxsbi/Hx8ejo6IjFixe3zIuI6Orqira2g60yPj4eY2Nj0d7eHkuWLDmuuSMjI9FsNmPJkiXR3n7w88harRYHDhyItra26OrqOq65o6Oj0Wg0YvHixdHRcfBbU6/XY//+/am5lUolTjjhhHLu/v37o16vR2dnZyxatCg9t9FoxOjoaEREnHjiieXcAwcORK1Wi0WLFkVnZ2d6brPZjJGRkYiIOOGEE47an5m5x7Lvp+I4mWx/TsVxcmh/vtnj5Mj9+WaPk9fbn2/2OJm4P9/scfJ6+/N4j5OJ+3Pij6ix0YNzOxYf3heH5u4/0Hp9dmx0JCKa0dG5JNr+bX/Wa7Wojx+ISqUtFi3papk7sm9R1BcXLft+bHTfUXNHR0djX7OtZX826vWoje0/au74/tFoNhvRvmhxtB8xN6ISnV2H9/34gf3RbNSjvaMz2g/tz8zcRiNqBw7u+86uw/uzNnYgGvVatHUsio5Fnem5zWYzxvcf3PeLlhzen7XxsWjUxnNz2zuio/Pwvp+4Pw/t+8zc+vh41GtjUWlrj0WLl0yYO8m+z8x9nePk0P7MzD2mff9mj5PX2Z+TzZ1qqSsQzebB/5H29vbGphe2xJaBXbHjtT1xy61/GkVRxFUf/4NybMdre+KUt70tiqKIJzY/X45Vv3hHFEURH/3d3yvHtgzsihUrVx6c++NN5djtf353FEUR//m/fLgc2zKwK8769XdEURTx8ONPlGN3fv0bURRFXPzBD5ZjWwZ2xX9817ujKIq4/8EflGPf/qu/jqIo4jcvfF/Les+/4L1RFEV8668fKMe+88CDURRFvPvc81rW+76L1kRRFPHNb99Xjn3vBw9HURTxG2ef3bLeD3zo0iiKIu7++vpy7KHH/3cURRG/duaZLev9yMd+J4qiiNu+/Gfl2I//z3NRFEWc1ttbju14bU9cdfV/jaIo4ub/8cVy7Ccv/L8oiiJ6enpa1vvxP7wuiqKIm/7klnLs2Zf/OYqiiKIo4pc7Xy1f+7pP3RhFUcQnb/xMOfbS9l+Vc599+Z/L8Zv+5JYoiiI+/ofXtaytp6cniqKIHz3/Yjn2p1/4UhRFEVf8/tUt3/fTenujKIp49KnN5dhtX/6zKIoiLvvo77R8HWf82plRFEU8tvFH5ft9Zd3XoyiK+K1LL235Xv7G2WdHURTxvR88XI5989v3RVEU8b6L1rSs99zzzo+iKOI7DzxYjv3V9/5XFEUR51/w3pb1/uaF74uiKGL9ffeXY/c/+IMoiiL+w7ve3bLeNRd/MIqiiHv+5zfLsQcfeTyKoohff8c7Wo6TD1324SiKIm7/87vLsSd+vCmKoogVK1e2rPd3f++qKIoiql+8oxx75vn/G0VRxClve1vLeq/6+B9EURTxuep/L8c2vbCl3J8T17v2+k9FURTx2c/dVI79dOv2cu7Pt+0o1/HJP/58FEUR13/qUy3f90NzD/2M2DKw67h/RuzctSf2D++N/cN74/ZL3hXV/r7Y/sJzsXNX68+Ia674aDlv//De+MqH3xvV/r7Y+tyT5djmv70/qv19ce8NV7TMvevKi+LMt5/U8jPie3/z3aj298VfrL2sZe6H/tOF5c+ILQO7YueuPfHSxoej2t8X6656f8vcb1z3kaj298Xz//BAOfbK0xuj2t8Xd37swpa53/mjq6La3xfPbLivHPvlT5+Jan9ffPm3z2uZe/9N10a1vy9+dP/6cuxfX/6nqPb3xRfef3bL3Adu/WRU+/viiW/dVY69+oufR7W/L2698MyWud//0uei2t8X//i128uxwZ3/EtX+vqj298XInt3l+N9/5dao9vfF33/l1nJsZM/ucu7gzn8px//xa7dHtb8vvv+lz7W8360XnhnV/r549Rc/L8ee+NZdUe3viwdu/WTL3C+8/+yo9vfFv778T+XYj+5fH9X+vrj/pmtb5n75t8+Lan9f/PKnz5Rjz2y4L6r9ffGdP7qqZe6dH7swqv198crTG8ux5//hgaj298U3rvtIy9x1V70/qv198dLGh8uxFx5/KKr9ffG1a36rZe5frL0sqv198dNHvl+Ovfzjx6La3xd3XXlRy9x7b7giqv19sflv7y/Htj73ZFT7++IrH35vy9y//ONrotrfF089cG85tv2F56La3xe3X/Kulrl/c/N/i2p/X2y8b13sH94bB/7tlygPncenQqWZeLXt27fHihUrpuzNAYC3zrZt2+L000+fktdKBUSj0YgdO3ZEs9mMlStXxrZt26K7u3tKFgIATJ2hoaFYsWJFbNu2LZYuXRp79+6N3t7e8mOgNyv1OxBtbW1x+umnx9DQUEREdHd3CwgAmMUOnat7enqm9HXdhQEApAkIACDtuAJi8eLFUa1WW27HAgBmj+k+V6d+iRIAIMJHGADAcRAQAECagAAA0gQEAJB2XAGxbt266OvriyVLlsTq1avj6aefnup1AQCT2LhxY1x66aXR29sblUolHnzwwZY/bzabccstt8Rpp50WXV1dsWbNmtiyZUvLnF27dsWVV14Z3d3dcdJJJ8UnPvGJGB4eTq0jHRDf/e5348Ybb4xqtRrPPvtsnHPOOXHxxRfHr371q+xLAQBJ+/bti3POOSfWrVs36Z/fcccd8dWvfjXuueee2LRpU5x44olx8cUXx/79+8s5V155ZbzwwgvxyCOPxEMPPRQbN26MtWvXptaRvo1z9erVcd5558Xdd98dEQefj7FixYq44YYb4vOf/3zqzQGA41epVGLDhg1x2WWXRcTBqw+9vb3xmc98Jj772c9GRMTg4GAsX7487r333rj88svjpZdeilWrVsUzzzwT5557bkRE/PCHP4xLLrkktm/fHr29vcf03qkrEGNjY7F58+ZYs2bN4Rdoa4s1a9bEk08+mXkpAGCKbd26NQYGBlrO0z09PbF69eryPP3kk0/GSSedVMZDRMSaNWuira0tNm3adMzvlQqIV199Ner1eixfvrxlfPny5TEwMJB5KQBgih06F/975+mBgYF4+9vf3vLnHR0dcfLJJ6fO5e7CAADSUgFxyimnRHt7e+zcubNlfOfOnXHqqadO6cIAgJxD5+J/7zx96qmnHnXjQ61Wi127dqXO5amA6OzsjPe85z3x6KOPlmONRiMeffTRuOCCCzIvBQBMsTPOOCNOPfXUlvP00NBQbNq0qTxPX3DBBbFnz57YvHlzOeexxx6LRqMRq1evPub36sgu7sYbb4yrr746zj333Dj//PPjzjvvjH379sU111yTfSkAIGl4eDheeeWVcnvr1q3xk5/8JE4++eRYuXJlfPrTn47bbrstzjrrrDjjjDPi5ptvjt7e3vJOjXe+853xgQ98IK699tq45557Ynx8PK6//vq4/PLLj/kOjIiIaB6Hu+66q7ly5cpmZ2dn8/zzz28+9dRTx/MyAEDS448/3oyIo/5z9dVXN5vNZrPRaDRvvvnm5vLly5uLFy9uXnTRRc2f/exnLa/x2muvNa+44opmURTN7u7u5jXXXNPcu3dvah0e5w0ApLkLAwBIExAAQJqAAADSBAQAkCYgAIA0AQEApAkIACBNQAAAaQICAEgTEABAmoAAANIEBACQ9v8BmyHfa7tyd54AAAAASUVORK5CYII=",
"text/plain": [
"<Figure size 640x480 with 1 Axes>"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"grades = lf.group_by('grade').len().collect().sort('grade')\n",
"min_grade = 0\n",
"max_grade = 100\n",
"df_0_100 = polars.from_dict({'grade': range(min_grade, max_grade+1)})\n",
"grades_0_100 = df_0_100.join(grades, on='grade', how='left').select(['len']).fill_null(0)\n",
"plot_error_bars_with_cutoff(grades_0_100.to_series().to_list(), min_grade, max_grade)"
"(csv_path, contributions) = _get_demo_csv_contrib()\n",
"assert csv_path == '/tmp/demo.csv'\n",
"assert contributions == 10\n"
]
},
{
Expand Down

0 comments on commit 4fdf1a5

Please sign in to comment.