Notebooks/day-14.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "02335352-e762-4d46-93d8-f0b6fe5dd905",
   "metadata": {},
   "outputs": [],
   "source": [
    "data = open(\"../inputs/day-14\").read().splitlines()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "1720179b-951f-44e5-8b29-e91675630264",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "772c1b6b-b64c-4033-9d5f-120f1008d1d5",
   "metadata": {},
   "outputs": [],
   "source": [
    "lines = [list(map(eval, line.split(' -> '))) for line in data]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "da155821-c68a-4ffc-98c0-0377bdd5ec81",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(528, 166)"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "max(max([l[0] for l in line]) for line in lines), max(max([l[1] for l in line]) for line in lines)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "09eaa03a-6c07-459c-baa0-e60c13d51c62",
   "metadata": {},
   "outputs": [],
   "source": [
    "grid = np.zeros((530, 200), dtype=int)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "5e770ccc-d767-4d7a-857f-56d4b3a2bf0b",
   "metadata": {},
   "source": [
    "0 - air, 1 - rock, 2 - sand."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "3a2baf88-3b57-4f5c-bb37-2389007c3292",
   "metadata": {},
   "outputs": [],
   "source": [
    "def build_line(line):\n",
    "    for i in range(len(line) - 1):\n",
    "        start, end = line[i], line[i+1]\n",
    "        x0, x1 = sorted([start[0], end[0]])\n",
    "        y0, y1 = sorted([start[1], end[1]])        \n",
    "        grid[x0:x1+1, y0:y1+1] = 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "08273a4b-b8bb-43dd-ae23-c017b2ac42ee",
   "metadata": {},
   "outputs": [],
   "source": [
    "for l in lines:\n",
    "    build_line(l)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "783f795d-fe80-4d71-ac8f-cc2d2705bd8e",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAH8AAAD8CAYAAABemXtlAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAAsTAAALEwEAmpwYAAAMn0lEQVR4nO2dW4xdVRnHf39bWgJiAIGmQrXFFBIgcaINmAgG5FaMWvAB2wcsl1iI9MFoogWNEnnxhkSjYobQUBIpEEilIQ2l9EE0gdgpNNxrL7ZhhtICIkU0QOHzYe+hZ6bnNmfvffY++/t+yeScvfa5rMnvrG+tfVnrk5kR+OQjZVcgKI+Q75iQ75iQ75iQ75iQ75jC5EtaKGmrpO2SVhT1PUHvqIjjfEnTgH8AFwKjwCZgiZk9n/uXBT1TVMs/E9huZjvN7F3gHmBRQd8V9Mj0gj73ROClhu1R4KxWL56hmXY4RxZUFd+8xRuvmdnxzfYVJb8jkpYBywAO5wjO0vllVaXWPGr37261r6iwPwbMadg+KS37EDMbNrMFZrbgMGYWVI2gHUXJ3wTMlzRP0gxgMbC2oO8KeqSQsG9mByQtB9YD04CVZvZcEd8V9E5hfb6ZrQPWFfX5QXbiDJ9jQr5jQr5jQr5jQr5jQr5jQr5jQr5jQr5jQn5BrH95S9lV6Ehpl3Q90O0P4OJPDBVaj1aE/IIoS+hUiLDvmJDvmAj7OdGqf69y+A/5OVFlya0I+Rlo19rbjfSr8kMJ+RloJ7EqgtvRs3xJc4C7gFmAAcNm9htJNwHfAl5NX3pjekvXwNOpNfe6v6wfSs/TtSTNBmab2ZOSjgI2A5cClwP/MbNfdftZH9OxFvftF8Ojdv9mM1vQbF/PLd/M9gB70udvSXqBZKZOrcjSmrNGiqLJZaKmpLnAY8AZwHeBK4H9wAjwPTN7o937o+UXRyEtfxxJHwUeAL5jZvsl3QbcTDIOuBm4Bbi6yfsmTNeqEkW25iodIWRq+ZIOAx4C1pvZr5vsnws8ZGZntPucaPnF0a7l93x6V5KAO4AXGsWnA8FxLgOe7fU7gmLJEva/AFwBPCNpS1p2I7BE0hBJ2N8FXJvhOyrHeGgehOP4TmQZ7f8NUJNdA3lMPwg3X+RNnOFLqUNLnipxSdcxblt+1U61loFb+Z4kt8KV/KmeYKn7D8SV/EG/BJs3tZffqkVHFChoBc6pEqd3i6PQCztVJM/WXuejgjjOd0yE/ZpT+7Cfdzj30gVEy685hVzPDwafkF8x1r+8pW+Xl0P+AFDUjyHkOybkDyB5RYKQ75jM8iXtkvSMpC2SRtKyYyVtkLQtfTwme1X9UfTAL6+Wf56ZDTUcT64ANprZfGBjuh1UjKLC/iJgVfp8FckEziAjeUeCPOQb8IikzekULIBZ6UROgFdIpnFPQNIySSOSRt7jnRyqEUyVPM7tn21mY5JOADZIerFxp5mZpEPOIZvZMDAMyendHOoRTJHMLd/MxtLHfcAakmyae8enbaWP+7J+T5A/meRLOjJdmAFJRwIXkczNWwssTV+2FHgwy/d4ZLx/b/WYB1nD/ixgTTJnk+nA3Wb2sKRNwH2SrgF2k6zWEXRBPy8PZ5JvZjuBzzQpfx2Ia7QVJ87wDSCNoT9LNxDyHRPyHRPyHVOLGzg90K5v7/XG0pA/QDST2UpwNwPBCPuOCfmOCfmOCfmOCfmOidH+AJH3nTwhf0Ao4mpfhH3HhPyKMjnEF3Ebd8h3TPT5FaJT6857lbCQXzFayWxWnrUryJJa7VTg3oaik4EfA0dT09RqdSPLevtbgSEASdOAMZJbt68Cbp1KarWgHPIa8J0P7DCz3Tl9XtAH8pK/GFjdsL1c0tOSVraaoRvTtcon82pckmYALwOnm9leSbOA1ziYWm22mR2SWq2RWI0roZcBXKfRftHr8F0CPGlmewHGHwEk3U6Sei3ogn6v55dH2F9CQ8iP1GqDQ6aWn87Pu5CJ6dN+UefUanlT5qqeWadrvQ18fFLZFZlq5Iwyl26NM3wlUJW1fUN+CUzlFG6RhPw+UcV0L3FJ1zGx5HrNqX2yhSrSrwQQWYiWX3Oi5feJdq2ziuleouXXnEizEjQl5OdAXgsk9ZuQ75gY8E2Bblt1p9dVJQ9fyJ8incR12l+lbiHCvmNCvmNCvmNCvmO6kp/ef79P0rMNZU0zaCnht5K2p/fuf7aoygfZ6Lbl3wksnFTWKoPWJcD89G8ZcFv2agZF0JV8M3sM+Nek4lYZtBYBd1nCE8DRk27nDipCluP8Vhm0TgReanjdaFq2hxpQpeP0rORykqdVBq12pGnYlgEczhF5VKNwqnJmLi+yjPZbZdAaA+Y0vO6ktGwCZjZsZgvMbMFhzMxQjaBXsshvlUFrLfDNdNT/eeDNhu4hqBDdHuqtBh4HTpU0mmbN+hlwoaRtwAXpNsA6YCewHbgd+Hbute4jvayTMyh01eeb2ZIWuw65/caSW4Ouz1KpqjHIgtsRZ/gcE5d0O1C3EX4j0fIdE/IdE/IdE/I7UNeRPoT8D+nHKtdVI+Q7JuQ7JuQ7JuQ7JuRPwsNAb5yQ75iQ3yV1jAghv4Fxwa0e60Zc1Uup89W7VkTLd0zIT8kztA9KN9Ex7EtaCXwF2GdmZ6RlvwS+CrwL7ACuMrN/S5oLvABsTd/+hJldV0TFi2BQpOVFN33+ncDvgLsayjYAN5jZAUk/B24AfpDu22FmQ3lWsh9En9+EZlO1zOwRMzuQbj5Bcm9+MGDkMdq/monJFedJegrYD/zIzP6aw3cMBO26jSpGlqxpVn4IHAD+lBbtAT5pZq9L+hzwZ0mnm9n+Ju8duOlanaii4HZkSad6JclA8Pz0Xn3M7B1IkuSZ2WZJO4BTgJHJ7zezYWAYkhU4e61H2Ux1+dTG/WXT06GepIXA94Gvmdl/G8qPT1OrIulkkjn6O/OoaJA/3RzqrQbOBY6TNAr8hGR0PxPYIAkOHtJ9EfippPeAD4DrzGzyvP5a0a4FV6F1t6Oj/BZTte5o8doHgAeyVqrKdArlve4v44cSq27XnFhvPyNZWnPWSFEk0fJrTqy3HzSl9vKncrHG28SN2ssPWhPym1D3Fj9OyMeP7MmEfMeEfMeEfMe4ke+1X2+HG/nBobiVH5HAofyQfpDaX9jJKrvqN2R0wvUl3UGXVyTuwn5wkI7yW2TWuknSmKQt6d+XG/bdkGbW2irp4qIqXjQervB10/Lv5NDMWgC3mtlQ+rcOQNJpwGLg9PQ9fxi/mzeoHj1N12rDIuAeM3vHzP5JknDhzAz1K4VWrbxurT9Ln788TZq4cjyhIq0zaw0ckweKdRw49ir/NuDTwBDJFK1bpvoBkpZJGpE08l4yySfoMz3JN7O9Zva+mX1AkkdnPLR3lVkr/YzIrlUyvU7XasyQeRkwfiSwFlgsaaakeSTTtf6erYpBUfQ6XetcSUOAAbuAawHM7DlJ9wHPk8zevd7M3i+k5gVTt8FdM2p/etc7cd9+0BS38puFdQ9n9RpxKz8I+a5xJ7+XUF7X8O9OfnCQkO+YkO+YkO+YkN+Eug7wJhPyHRPy8dPSJ+Puwk6vogf1Th7X9+1PZlAlFkGEfceEfMeEfMeEfMf0Ol3r3oapWrskbUnL50r6X8O+PxZY9yAjPWXXMrNvjD+XdAvwZsPrBzK7lke6WW//sTRf3iEoybRwOfClnOsV9IGsff45wF4z29ZQNk/SU5L+IumcjJ8fFEjWkzxLgNUN266zaw0aPbd8SdOBr9OQUy+dnft6+nwzSarVU5q9P6ZrlU+WsH8B8KKZjY4XRHatwaKbQ73VwOPAqZJGJV2T7lrMxJAPSXatp9NDv/txkF1rkOk1uxZmdmWTstpn16oTcYbPMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMSHfMZVYkEnSq8DbwGtl16UAjqPc/+tTZnZ8sx2VkA8gaaTVqlGDTJX/rwj7jgn5jqmS/OGyK1AQlf2/KtPnB/2nSi0/6DOly5e0UNJWSdslrSi7PllJVyd7Jl2NbCQtO1bSBknb0sdjOn1OPyhVfrqQw++BS4DTgCWSTiuzTjlxnpkNNRzirQA2mtl8YGO6XTplt/wzge1mttPM3gXuARaVXKciWASsSp+vAi4tryoHKVv+icBLDdujadkgY8Ajkjani04BzDKzPenzV4BZ5VRtIu6WXO8DZ5vZmKQTgA2SXmzcaWYmqRKHWGW3/DFgTsP2SWnZwGJmY+njPmANSde2V9JsgPRxX3k1PEjZ8jcB8yXNkzSDZJGntSXXqWckHSnpqPHnwEXAsyT/09L0ZUuBB8up4URKDftmdkDScmA9MA1YaWbPlVmnjMwC1iSr0jIduNvMHpa0CbgvXclsN8mStaUTZ/gcU3bYD0ok5Dsm5Dsm5Dsm5Dsm5Dsm5Dsm5Dvm/6P0HMAY2tNAAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "plt.imshow(grid[450:,:].T, interpolation='nearest')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "355862aa-c43f-4d24-ae61-ea3570f147cc",
   "metadata": {},
   "source": [
    "When a sand gets to the last layer at `grid[:,-1]` it means we're done. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "f3567b30-a4d1-4e47-a6fb-5b644231ad55",
   "metadata": {},
   "outputs": [],
   "source": [
    "total = 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "ebc48f0b-42a9-4331-a247-4366124e803f",
   "metadata": {},
   "outputs": [],
   "source": [
    "def drop_sand():\n",
    "    global total\n",
    "    sand = (500,0)\n",
    "    while True:\n",
    "        if sand[1] == 167:\n",
    "            return\n",
    "        if grid[sand[0], sand[1] + 1] == 0:\n",
    "            sand = sand[0], sand[1] + 1\n",
    "        elif grid[sand[0] - 1, sand[1] + 1] == 0:\n",
    "            sand = sand[0] - 1, sand[1] + 1\n",
    "        elif grid[sand[0] + 1, sand[1] + 1] == 0:\n",
    "            sand = sand[0] + 1, sand[1] + 1\n",
    "        else:\n",
    "            grid[sand] = 2\n",
    "            total += 1\n",
    "            return True"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "efc81f6a-a85f-4006-a865-20f52012b61b",
   "metadata": {},
   "outputs": [],
   "source": [
    "while drop_sand():\n",
    "    pass"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "ad8e9562-3404-4c8d-b5e4-9e5dbb672445",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1003\n"
     ]
    }
   ],
   "source": [
    "print(total)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "9546a81f-69a5-440d-b324-fef146fe5f6a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAH8AAAD8CAYAAABemXtlAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAAsTAAALEwEAmpwYAAAO8ElEQVR4nO2df6wlZXnHP9/yY4loo0TcUKCy4koCJN7UDZIUjRZBMLWrbSK7f1RQ40oq/9gmLdqmGo2J/WFv2tRK1nQDJHVZgqVuCXFBmkibuHHv6kZA2HLvFsLeXXeFWtG2AVYf/5g57NyzM3PmzMx7Zua8zyfZnHNmzpl5737f53nfeX88j8wMJ05+pesCON3h4keMix8xLn7EuPgR4+JHTDDxJV0n6aCkZUm3hrqPUx+FeM6XdBrwn8A1wGFgH7DVzH7Q+s2c2oSy/CuAZTM7ZGYvAncBmwPdy6nJ6YGuez7wTObzYeCtRV8+U+vsLM4OVJS4+Sk/ftbMzs07F0r8iUjaBmwDOItX8FZd3VVR5ppv2j1PF50L5fZXgQszny9Ij72MmW03s01mtukM1gUqhlNGKPH3ARslbZB0JrAF2B3oXk5Ngrh9Mzsh6RZgD3AasMPMHgtxL6c+wdp8M7sfuD/U9Z3m+AhfxLj4EePiB2LPkQPsOXKg62KU4uIHICt6nyuBi98yRUL3sQK4+C0ySeC+VQAXvyWqCtunCuDit8C0gvalArj4DakrZB8qgIvfgDIBL951c6Pfz4LOpnSHThXhqlQAFuGNn9jbvEA1cPGnpKq1rtxwG+/+tYWgZWmKu/2IcfGnYKi9+iLc7VdkkpBF7fvKkdsAetkEuPgVqGLBeW38+Bh/XgUYfaeLyhFk3f60/KrOsb4u4KzzOPfGT+xlefHKwt+Nevfj1w5RAb5p9+w3s01551z8EqpYfF3Biq7ddgUIIr6kC4E7gfWAAdvN7G8lfQb4KPCj9KufSpd0FdJH8fPEKXtur2LtRedXbrjtlGNtVYJQ4p8HnGdm35X0KmA/8D7gA8DPzOyvq16rb+KXWXzblhn6XmXi1+7wmdlR4Gj6/qeSHifZqTNoxsU4xdoXT74ts+aqnuDiXcXeoKiT2BattPmSLgIeBi4H/hC4CXgeWAL+yMx+XPb7vlj+rNrhacvQ5P5BO3ySXgl8C/i8mf2zpPXAsyT9gM+RNA0fzvlddrvWW67SexqVoynZ//Qmvfi656s+IUxLMPElnQHcB+wxs7/JOX8RcJ+ZXV52na4tv21r6xNl4tce3pUk4B+Bx7PCpx3BEe8HHq17DycsTUb4fhP4feARSQfSY58CtkpaIHH7TwEfa3CPoJT1tIs6WyPX3NU0bJs06e3/B6CcU4PYojWxVw9revbzSJRj+3kWP4T597aJbkp3kquPiagsv8rj3MqReDxANOKPW3XZFGzokbW+EIX4I1GLhmpzh1o7XFg5K+Z+SncWc+Z9JsjEzhDYc+RArrXXGWadRy8wt5aftfjYrD1LdJZf1H7Xsfay3wyd6J7znZPMrdt3Eube7bftzmNpAtzy55wg8/nO8HHxe8by4pWnNC+hInq5+D0nO1DVdgVw8XtMnthtVgAXv6eUiVy2yncaXPyIaSy+pKckPSLpgKSl9Ng5kh6U9GT6+prmRY2PSjF9GtCW5b/TzBYyz5O3Ag+Z2UbgofSz0zNCuf3NwB3p+ztINnA6DWnbE7QhvgEPSNqfbsECWJ9u5AT4Ick27jVI2iZpSdLSS7zQQjGcaWljbP8qM1uV9DrgQUlPZE+amUk6ZQzZzLYD2yEZ3m2hHM6UNLZ8M1tNX48D95Jk0zw22raVvh5veh+nfRqJL+nsNDADks4GriXZm7cbuDH92o3A15vcJ0ZG7XvRaxs0dfvrgXuTPZucDnzVzL4haR9wt6SPAE+TROtwKjDL6eFG4pvZIeDNOcefA3yONiBt7C3wEb4BMhre3XPkQKOhXhc/Ylz8geGzepHS9nz+XCzgjIGysG11F5a6+AMiL2ZvUVCJKh1Bd/sR4+IPjKKlXXX6Ay7+gGg7Y6eLP2dMUwFc/AEQKreP9/Z7TpUgUnVx8XtMlUe6Jrjb7ymhN2yAi987Ro9t4y4+xJYtF79HVBW2rQrgbX5PyI0cNsaa85kYgnVx8XtA1pKLcu7ldfia7tmrLb6kS4BdmUNvAP4ceDVTplZzuqFJvP2DwAKApNOAVZKl2x8CFqdJrRYz4xZeJwFEXdrq8F0NrJjZ0y1dLwrqdNz62NvfAuzMfL5F0vcl7SjaoRv7dq0mIrZVAdpIrXYmcAS4zMyOVU2tliW2aFxF4tUZvp3U2w8dh+964Ltmdgxg9Aog6SskqdeclDKrnXWqlzbc/lYyLt9TqxVTxV2HiryVRyPLT/fnXcPa9Gl/OZTUarNkWlc/i3QvHoEzMFWsOCty28kh5j72bl+pbO2LJVm1A6Z7ccsPxCSLn1XHzi1/xuQJ38d0Lz6lGzHu9lumzN13kevH3f6MmLjYssTVd9EEuOW3RJ7F9yGrl1t+YMoydo6ss4/pXtzyG9L3jJ2eZiUQs1heHRIXvyZFnbuLd908mArg4k9J1Vm3IVQA7/BNQVlolCwve4LF/PN9ycPn4k9JnnDT7KlrK0VKG7jbb8iQO30ufsS4+A2YtMa+77j4NZmHHn8l8dP198clPZo5lptBSwl/J2k5Xbv/G6EK3wXTLrCc5YLMaalq+bcD140dK8qgdT2wMf23Dfhy82IOnz5WgErim9nDwH+PHS7KoLUZuNMS9gKvHlvOHS19qwBNnvOLMmidDzyT+d7h9NhR5oDQiQ5nSSuDPEUZtMpI07BtAziLV7RRjOD0ZWSuLZr09osyaK0CF2a+d0F6bA1mtt3MNpnZpjNY16AYs6FvLrsNmohflEFrN/DBtNd/JfCTTPMwSEbCz1sFqPqotxP4NnCJpMNp1qwvANdIehJ4V/oZ4H7gELAMfAX4g9ZLPUPGx+LHH936NFY/LZXafDPbWnDqlOU3liwN+niTQvWNosWYQ8dH+CLGp3RLSGLgdF2KcLjlFzCvnbwsLn4O00TIGjIu/hhFQs9jBXDxM0zaXTtvFcDFTxl/Xi8aw+/zFO20uPg1mYcK4OI3YOgVwMVvyJArgIvfAkOtAC7+GHUXawyxArj4EePiV6SKRxia9XtwhpQq8/IrN9xW6Vp9CtDgYVkqMGl93rRr9ftUAYpwt1+BrjNihMLFTyly+3VEDJEAMQQT3b6kHcBvA8fN7PL02F8B7wVeBFaAD5nZ/0i6CHgcOJj+fK+ZDWahe14FqBKMoYxRBehjMzCxwyfp7cDPSHbhjMS/Fvg3Mzsh6S8AzOxPUvHvG32vKn3o8LVJkcX3LQLnRLeft1XLzB4wsxPpx70ka/MdhrVtu43e/odZm1xxg6TvAc8Df2Zm/97CPQZBWcye0WNin54EmqZZ+VPgBPBP6aGjwK+b2XOS3gL8i6TLzOz5nN8ObrvWJCbF68ke60MFaJJO9SaSjuDV6Vp9zOwFSJLkmdl+SSvAm4Cl8d+b2XZgOyRtft1ydM2k8KmFnmAGOXQmUetRT9J1wB8Dv2Nm/5c5fm6aWhVJbyDZo3+ojYI67VOlt78TeAfwWuAY8Gngk8A64Ln0a3vN7GZJvwd8FngJ+AXwaTP710mF6Kq3n3XJs7TCWd63rLcf7dh+3RDpZXMAZRGxJ50Ptf3bxR+jb9kwQuITOxkmPWvn9cSbWHNTTxGSqCx/mkGWefEAHm+f6UfX+jYaF4K5F3958crKQo6v1hly4IUqzL34TjEufg7zFG6tDBefeMQex8WPGBc/Ylz8iIlG/Fjb9TKiEd85lWjFd08Qofgu+knmfmKn6RDt0MOsRz2l21S8viy2DEF0bn8a5j0K50TxCzJrfUbSqqQD6b/3ZM59Ms2sdVDSu0MVPDTjzcU8zvBVsfzbOTWzFsCimS2k/+4HkHQpsAW4LP3NP4xW8zr9o9Z2rRI2A3eZ2Qtm9l8kCReuaFC+Tiiy8nmz/iZt/i1p0sQdo4SKFGfWGhzjUTiqRuUYEnXF/zJwMbBAskXri9NeQNI2SUuSll5KNvk4M6aW+GZ2zMx+bma/IMmjM3LtlTJrpdfoZXatee3Z51F3u1Y2Q+b7gdGTwG5gi6R1kjaQbNf6TrMizo6YhIdqkTle3q4l6TDJdq13SFoADHgK+BiAmT0m6W7gByS7dz9uZj8PUvKWGRc+hmHguR/erULdrVtDwNftF1AWOz+GJiBa8fODL/m6fScSohQ/BpdehejEr+PK59X9Rye+cxIXP2Jc/Ihx8SPGxc8hhqFdcPGjxsUnHksfJ7qJnbrP7ENdvx/1uv1xhipiCNztR4yLHzEufsS4+BFTd7vWrsxWrackHUiPXyTp/zPn5m+x+xxRpbd/O/D3wJ2jA2Z2w+i9pC8CP8l8f8XMFloqnxOQieKb2cNpyrRTkCTgA8BvtVwuZwY0bfPfBhwzsyczxzZI+p6kb0l6W8PrOwFpOsizFdiZ+Rx1dq2hUdvyJZ0O/C6ZnHrp7tzn0vf7SVKtvinv933drhUTTdz+u4AnzOzw6IBn1xoWVR71dgLfBi6RdFjSR9JTW1jr8gHeDnw/ffS7B7jZzKru7XdmTJXe/taC4zflHPsa8LXmxXJmgY/wRYyLHzEufsS4+BHj4keMix8xLn7EuPgR4+JHjIsfMS5+xLj4EePiR4yLHzEufsS4+BHj4keMix8xLn7EuPgR4+JHjIsfMb0IyCTpR8D/As92XZYAvJZu/67Xm9m5eSd6IT6ApKWiqFFDps9/l7v9iHHxI6ZP4m/vugCB6O3f1Zs235k9fbJ8Z8Z0Lr6k6yQdlLQs6dauy9OUNDrZI2k0sqX02DmSHpT0ZPr6mknXmQWdip8GcvgScD1wKbBV0qVdlqkl3mlmC5lHvFuBh8xsI/BQ+rlzurb8K4BlMztkZi8CdwGbOy5TCDYDd6Tv7wDe111RTtK1+OcDz2Q+H06PDRkDHpC0Pw06BbDezI6m738IrO+maGuJLuT6DLjKzFYlvQ54UNIT2ZNmZpJ68YjVteWvAhdmPl+QHhssZraavh4H7iVp2o5JOg8gfT3eXQlP0rX4+4CNkjZIOpMkyNPujstUG0lnS3rV6D1wLfAoyd90Y/q1G4Gvd1PCtXTq9s3shKRbgD3AacAOM3usyzI1ZD1wbxKVltOBr5rZNyTtA+5OI5k9TRKytnN8hC9iunb7Toe4+BHj4keMix8xLn7EuPgR4+JHjIsfMb8ERW1GXD7EAAsAAAAASUVORK5CYII=\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "plt.imshow(grid[450:,:].T, interpolation='nearest')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "6701975d-9b7a-4516-bf78-ba60a43d2b99",
   "metadata": {},
   "source": [
    "## Adding a floor"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b3c2d6cd-a882-4ef7-9df9-032657b40e19",
   "metadata": {},
   "source": [
    "Need bigger grid:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "19959f6d-6f2a-4118-aad3-62692b1b92d2",
   "metadata": {},
   "outputs": [],
   "source": [
    "grid = np.zeros((750, 200), dtype=int)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "d92013da-deb8-44df-8109-ed0d037fee1c",
   "metadata": {},
   "outputs": [],
   "source": [
    "for l in lines:\n",
    "    build_line(l)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "3b26fc72-32d5-44bc-8b7e-f5c509e00e7a",
   "metadata": {},
   "source": [
    "Adding a floor:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "c8cb66d1-7373-4096-be03-91f6e778459f",
   "metadata": {},
   "outputs": [],
   "source": [
    "grid[:, 168] = 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "1d0ef22b-426f-4683-999e-678f20497a18",
   "metadata": {},
   "outputs": [],
   "source": [
    "total = 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "614ae806-1ba1-4c00-88bb-5c14a432a60d",
   "metadata": {},
   "outputs": [],
   "source": [
    "def drop_sand():\n",
    "    global total\n",
    "    sand = (500,0)\n",
    "    while True:\n",
    "        if grid[sand[0], sand[1] + 1] == 0:\n",
    "            sand = sand[0], sand[1] + 1\n",
    "        elif grid[sand[0] - 1, sand[1] + 1] == 0:\n",
    "            sand = sand[0] - 1, sand[1] + 1\n",
    "        elif grid[sand[0] + 1, sand[1] + 1] == 0:\n",
    "            sand = sand[0] + 1, sand[1] + 1\n",
    "        else:\n",
    "            grid[sand] = 2\n",
    "            total += 1\n",
    "            if sand == (500,0):\n",
    "                return\n",
    "            return True"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "id": "23010eee-b7b2-40b5-af57-2787b06e13a7",
   "metadata": {},
   "outputs": [],
   "source": [
    "while drop_sand():\n",
    "    pass"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "id": "9af562c3-ea65-4c4c-91db-324b3d84dabb",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "25771\n"
     ]
    }
   ],
   "source": [
    "print(total)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "id": "4d8fb472-59d3-4b0e-aee7-64f200831da6",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAXcAAADKCAYAAABe4wDhAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAAsTAAALEwEAmpwYAAAWrklEQVR4nO3df7BcZX3H8ffXGEIFW4wymfDDhvzQjlgbMcV0RFoF5ce0BqxDQmcUf4yRFmZqRkcT7SjVYUatmLFjayaOFLAthOIvStGI6GidacCgIQQxcIOx5IdEQQG1EwG//eM8m5y7Obt39/zY85xzPq+ZO3f32b273/uc3c997nPOnsfcHRERaZdn1F2AiIiUT+EuItJCCncRkRZSuIuItJDCXUSkhRTuIiItVFm4m9m5ZrbTzKbMbG1VzyMiIkeyKo5zN7NZwP3Aa4A9wHeBi939B6U/mYiIHKGqkfvpwJS7P+juvwFuAFZU9FwiItLnmRU97onAQ6nre4CXD7rzUTbHj+aYikoRye8FL/k1929/Vt1liGR6gp//zN2Pz7qtqnCfkZmtBlYDHM2zeLmdVVcpIgNt3rwNgHNOWFprHSJZvu43/XjQbVVNy+wFTk5dPym0HeLuG919mbsvm82cisoQyW/zvm2Zl0WaoKqR+3eBJWZ2CkmorwL+qqLnEinVoCDvtWsUL01Qycjd3Z8CLgc2A/cBN7r7vVU8l0iZRhmhaxQvTVDZnLu73wrcWtXji5RNoS1tok+oiuSgPwQSO4W7CPnCWgEvMavtUEiRGBQNaO1klVhp5C6dVebIW6N4iY3CXTpJYSxtp3AXKYn+YEhMFO7SOVWGsAJeYqEdqtIZeYJ32I5SfZJVYqaRu3RC3hH1TAFexXOKlEEjd2m9cUJ20aZLj2jbtW+DRuHSOAp3kZRdK6cHee8Pw+Z92w61j/PHIv1zIpOkcJdWGzWIp43Y1yffFq/ZktkOoz1mugYFvExa7jVUzexk4DpgHuDARnf/pJldAbwd+Gm46/vCScQG+l2b61qsQ8o0SqhnTcEsXrOFqfXLR27ftXLDWHUp5KVMX/eb7nL3ZVm3FQn3+cB8d/+emT0buAu4ALgI+KW7f3zUx1K4S5nGmTYpErZ5d5gq4KUsw8I997SMu+8H9ofLT5jZfSRrp4rUZuwRe2oKJmtkPvxxpt9/3FG8SJVKmXM3swXAS4E7gFcAl5vZm4CtwLvc/edlPI9IGfp3mvYsXrOl2AOvHO1umoOXScg9LXPoAcyOBb4FXOnuXzCzecDPSObhP0wydfPWjJ9LL5D9sjPs/EJ1iAwbtZcxv172fLwCXoqqZM4dwMxmA7cAm939Exm3LwBucfcXD3sczblLUaNMx0wyTEedj1fASxGVzLmbmQGfBe5LB7uZzQ/z8QAXAjvyPofITGYK0exDGacrPB0zRj2D7q+Ql7IVmXN/BfBG4B4z2xba3gdcbGZLSaZldgPvKPAcIgONEqSD5teroNMNSEyKHC3zHcAybtKi2CJj0k5WKZs+oSqNNM7O00mcG6aMUbsCXspU+GiZMmiHqoxj3J2nVc9rlz0do4CXUVWyQ1Vk0vIeEdOUUO9/XIW8FKFwl0YYFKT9UzCLOfKY86zj0IseIaOdpxI7TctIIwwL0zpGuJMKd43eZRhNy0ijZQVp//Hrg0bnVYziJzlq105WyUsjd4laV0fs/RTwkkUjd2mcQSFaV8jVPceunawyLi2QLdEZFqR1hGzdwZ4WUy0SN4W7RGWU8Jr0nLdIEyncpZFiD92sUwyXJfbfXeKgcJdoxBZasdWTFnNtEofCO1TNbDfwBPA08JS7LzOzucAmYAHJmSEv0mpMMkjeoKpqJ2PZwblo06WVLMGnnawyTFkj91e5+9LUITlrgdvdfQlwe7gucoSyTrhVlqKPVeV0zCAaxUuWqqZlVgDXhsvXAhdU9DzSYDGFclmPIRKLMsLdga+Z2V1hXVSAeanVmH4CzCvheUSG6nI4d/l3l2xlhPsZ7n4acB5wmZmdmb7Rk4/AHvExWDNbbWZbzWzrkxwsoQxpktjCKLZ68mjD7yDlKRzu7r43fD8AfBE4HXjYzOZDsqYqcCDj5za6+zJ3XzabOUXLkAapMoTyPHabQrFNv4sUU+jcMmZ2DPAMd38iXL4N+BBwFvCIu3/EzNYCc939PYMeR+eW6YZJB89MR5FMapENnY9GqjLs3DJFR+7zgO+Y2d3AncB/uftXgY8ArzGzB4Czw3XpsNhOG1BFPVmPqZG01KVQuLv7g+7+R+HrVHe/MrQ/4u5nufsSdz/b3R8tp1yR8Uw6XGc6PfEkD5XUH5Zu0ydUpXKxhUzV9fQeP4bfO4YapB4Kd6lUDOGSrmFS9cTwe/fEVItMjs7nLpWILVDqqGfQFEx/exWnJuinUxV0j8JdShdbsNdp8Zot0/pj18oN0wK2f8k/kbJoWkakQrH9oYutHqmOwl1KpfCYWd19VPfzy2Qo3KU0Co3R1d1XdT+/VE9z7lKYgmKwoce1r59cHVm0k7XdFO5SiIJ9sP6dp1nUf1IVTcuI1CSWYI+lDimXwl1yUyhM1z8Fs2jTpQP7KLa+i60eKU7hLrkoDEbX31ex9l2sdUk+uefczeyFJItg9ywEPgAcB7wd+Glof5+735r3eSQuCoBsWYtgZ31gaRKfRi1CO1nbI3e4u/tOYCmAmc0C9pIs1vEWYL27f7yMAiUeCnaR5ihrWuYsYJe7/7ikxxNphab+QWxq3XJYWeG+Crg+df1yM9tuZleb2XNKeg6pkd7s+TW175patyQKH+duZkcBrwPWhaZPAx8mWRT7w8BVwFszfm41sBrgaJ5VtAypkN7ko4n5A0t5bd63TfPvDVVoDVUAM1sBXObur824bQFwi7u/eNhjaA3VeCnYR9fmDywp4OM0bA3VMj6hejGpKRkzm+/u+8PVC4EdJTyHTFiTg2hSjjgv+77sT6S2oS81gm+eQiN3MzsG+F9gobs/Fto+R3IUjQO7gXekwj6TRu5xaUMYTUo68LIOI2xbXyrg41LZyN3dfwU8t6/tjUUeUyR200bs65MFOabWL2fRpuXT2tpII/jmKDznXgaN3OPRtpFmVUYJuDb3pQI+DlXPuUtLtDmMiuqfX1/MliOWyOuN4A//TPyfSM1LI/j4aeQuCvUhjgj1NTOH+qC2tga9Qr4+w0buCveOU7APV1ZwtbmfFe710bSMyBgG7TBNG7ft0M5W2jeC1xRNnDRy77A2jyaLqjKs2trvCvjJGzZy1/ncO6qtAVOWqvqnzf3e5t+tiRTuHbN53za9CQdIT8cMW0Upj670e1d+zybQnHuH6E033dATffVuzzjh17gfUFK/Sx0U7tJp6aBOh3D/Kko9/TtMZ9LVYNdO1vppWqYjuhoyRajPilH/1Uvh3gF6k82sij5Sv6sP6jTStIyZXQ38OXCgd252M5tLskD2ApKzP17k7j83MwM+CZwP/Bp4s7t/r/zSZSZ6Y81spj7Ks2C0+n06Lbpdj1FH7tcA5/a1rQVud/clwO3hOsB5wJLwtZpkZSaZMAVMuUbtT/W7xGKkcHf3bwOP9jWvAK4Nl68FLki1X+eJLcBxZja/hFpFajXqKF+yqX8mq8jRMvNSi3D8BJgXLp8IPJS6357QNnTBDimP3kSjm+lwSCmXjqKZnFIOhXR3N7OxzmOgBbKroWAf3aDDHfNQv49OAT8ZRY6Webg33RK+Hwjte4GTU/c7KbRN4+4b3X2Zuy+bzZwCZUiPAmZ8ZfSZ+n186rPqFQn3m4FLwuVLgC+n2t9kieXAYzOtoSrF6CPf4+mfisnbf+r3YtR31Rr1UMjrgT8Dnmdme4APAh8BbjSztwE/Bi4Kd7+V5DDIKZJDId9Scs0ihWXNtS9m9NMKKJjKoSma6uiUvw2nkMmnaKCo38ulgM9Hp/xtKQVMPdTv5VOflk/h3lB6MxSTt//U79VR35ZL4d4w2olXXG++fZy+VL9Phvq4PAp3abX+BTiy6JOncVF/l0Ph3iB60VdnUN+qz+uhfi9O4d4QerFL1+g1X4zCvQH0Ip+M/n5Wv9dP2yA/LbMXMb2wy7Fo06XsWrlhpPuqz+OjDzrlo5G7dJLOBtks+qM7PoV7pPRiLlcvzBXqzaX3xHg0LRMhvYjLM+i0vurjZtIUzeg0co+MQkdkOL1HRjPjicMGLI79D8BfAL8BdgFvcfdfmNkC4D5gZ/jxLe4+4//BOnGYXrBVKToNM+qOWKlH10fxw04cNsq0zDXAp4DrUm23Aevc/Skz+yiwDnhvuG2Xuy/NX273KNirU2S1JW0XabIZp2WyFsd296+5+1Ph6haS1ZZEWkPB3gzaToOVsUP1rcCm1PVTzOz7wOPA37n7f2f9kNZQTejFWb1xd8JNrV/Ook3Lp7VpeiZe2smabaTFOsJc+i29OfdU+/uBZcDrwyLZc4Bj3f0RM3sZ8CXgVHd/fNjjd3XOXcE+WZqeabcuBnzROfdMZvZmkh2tZ3n4C+HuB4GD4fJdZrYLeAGwNe/ztJUCY/JmGuFNrV+e2d4/igeN5GOkEfx0ucLdzM4F3gP8qbv/OtV+PPCouz9tZguBJcCDpVTaEgr1evX6vxcCWYG+eM2WGdvTR+Eo6OOhgD9slEMhDy2ODTxMsjj2OmAO8Ei42xZ3v9TM/hL4EPAk8Fvgg+7+nzMV0aVpGYV7HMoKAG3POHUl4IdNy2iB7AlSEMRj1OPfRxnFp2kUH48uBLzCPQIK9rhU8cbXNo5P2wN+WLjr9AMToDd9fLRNuqHL21nhXiEtqly9cU8vkL5/WdtH2zluXd02CnfphGF/BIq8+bsaHE3Txe2kcK9IF19MsRlnVJ9ne2kbN0vXtpfCvQJdexF1kbZxM3VpuyncS9alF0/baNt1Q1e2s8K9JNqpVq+yls+baTtqO7dDF7ahwl1ap4ygz3rzdyEQuqTt21PhXoK2v0i6Kr1dtY3bqc3bVZ9QLajNL44myDNKH+cUAeecsFTbuAOa+klWnX6gInrTx2GcN6a2mQzSxIAvdPoBM7vazA6Y2Y5U2xVmttfMtoWv81O3rTOzKTPbaWbnlPMrxEU71eIyyvbYvG/btBN9lbUDVtqjbe/pUebcrwHOzWhf7+5Lw9etAGb2ImAVcGr4mX82s1llFSsyzKA3Z9vetFKdNr1Wci2QPcQK4AZ3P+juPwKmgNML1BedNm38LskaqWv0Llna8h4vcrTM5Wa2PUzbPCe0nQg8lLrPntB2BDNbbWZbzWzrk8nKfNFry0Zvs/5tlL6e3pGq867LMG14r+cN908Di4ClwH7gqnEfwN03uvsyd182mzk5y5icNmzsruhtK20zKaLpr59c4e7uD7v70+7+W+AzHJ562QucnLrrSaGt0Zq+kbtI20zK0OTXUd4Fsue7+/5w9UKgdyTNzcC/m9kngBNIFsi+s3CVNWryxpXDdq3ccMShbrv2aWpGZtbURbdHORTyeuB/gBea2R4zexvwMTO7x8y2A68C1gC4+73AjcAPgK8Cl7n705VVXzEFe7voE6eSVxNfL6McLXOxu89399nufpK7f9bd3+juf+juL3H316VG8bj7le6+yN1f6O5fqbb86jRxY0pi2FEwWYtai4yiaZmgc8tkaNpGlPFo+0peTXrtKNz7NGnjicjkNSUjFO6BTikgIqNqQlYo3EVEcog94BXuxL+RRCROMWdH58M95o0j5dF5ZKQqsWZIp8M91o0iIs0SY5Z0crGOGDeElGPUEbpOHCZVmPQnWYct1pHr9AMisVq8Zstod1xZbR3STTGdqiCKkfuc55/sJ7zrnXWXISLSKLvf+e64R+5zHvrV6CMuEREBYPeQ2zq9Q1VEpK3yLpC9KbU49m4z2xbaF5jZ/6Vu014rEZEajDItcw3wKeC6XoO7H9odZWZXAY+l7r/L3ZeWVJ+IiOQwY7i7+7fNbEHWbWZmwEXAq0uuS0RECig65/5K4GF3fyDVdoqZfd/MvmVmrxz0g01cIFtEpCmKHi1zMXB96vp+4Pnu/oiZvQz4kpmd6u6P9/+gu28ENkLyIaaCdYiISErukbuZPRN4PbCp1+buB939kXD5LmAX8IKiRYqIyHiKTMucDfzQ3ff0GszseDObFS4vJFkg+8FiJYqIyLjyLpANsIrpUzIAZwLbw6GRNwGXuvujJdYrIiIjGOVomYsHtL85o+3zwOeLlyUiIkXoE6oiIi2kcBcRaSGFu4hICyncRURaSOEuItJCCncRkRZSuIuItJDCXUSkhRTuIiItpHAXEWkhhbuISAsp3EVEWkjhLiLSQgp3EZEWUriLiLSQude/fKmZ/RT4FfCzumsZwfOIv84m1Aiqs2yqs1xNqPP33f34rBuiCHcAM9vq7svqrmMmTaizCTWC6iyb6ixXU+ocRNMyIiItpHAXEWmhmMJ9Y90FjKgJdTahRlCdZVOd5WpKnZmimXMXEZHyxDRyFxGRktQe7mZ2rpntNLMpM1tbdz1pZrbbzO4xs21mtjW0zTWz28zsgfD9OTXUdbWZHTCzHam2zLos8Y+hf7eb2Wk113mFme0NfbrNzM5P3bYu1LnTzM6ZUI0nm9k3zewHZnavmf1taI+qP4fUGVt/Hm1md5rZ3aHOvw/tp5jZHaGeTWZ2VGifE65PhdsX1FznNWb2o1R/Lg3ttb2PcnP32r6AWcAuYCFwFHA38KI6a+qrbzfwvL62jwFrw+W1wEdrqOtM4DRgx0x1AecDXwEMWA7cUXOdVwDvzrjvi8L2nwOcEl4XsyZQ43zgtHD52cD9oZao+nNInbH1pwHHhsuzgTtCP90IrArtG4C/Dpf/BtgQLq8CNk2oPwfVeQ3whoz71/Y+yvtV98j9dGDK3R90998ANwAraq5pJiuAa8Pla4ELJl2Au38beLSveVBdK4DrPLEFOM7M5tdY5yArgBvc/aC7/wiYInl9VMrd97v798LlJ4D7gBOJrD+H1DlIXf3p7v7LcHV2+HLg1cBNob2/P3v9fBNwlplZjXUOUtv7KK+6w/1E4KHU9T0Mf8FOmgNfM7O7zGx1aJvn7vvD5Z8A8+op7QiD6oqxjy8P/9penZrWqr3OMCXwUpJRXLT92VcnRNafZjbLzLYBB4DbSP5r+IW7P5VRy6E6w+2PAc+to0537/XnlaE/15vZnP46gxjeR0PVHe6xO8PdTwPOAy4zszPTN3ry/1p0hxvFWlfwaWARsBTYD1xVazWBmR0LfB54p7s/nr4tpv7MqDO6/nT3p919KXASyX8Lf1BvRdn66zSzFwPrSOr9Y2Au8N76Kiym7nDfC5ycun5SaIuCu+8N3w8AXyR5oT7c+3csfD9QX4XTDKorqj5294fDm+q3wGc4PFVQW51mNpskMP/N3b8QmqPrz6w6Y+zPHnf/BfBN4E9IpjGemVHLoTrD7b8HPFJTneeG6S9394PAvxBRf46r7nD/LrAk7Ek/imSHys011wSAmR1jZs/uXQZeC+wgqe+ScLdLgC/XU+ERBtV1M/CmsLd/OfBYarph4vrmKS8k6VNI6lwVjp44BVgC3DmBegz4LHCfu38idVNU/Tmozgj783gzOy5c/h3gNST7B74JvCHcrb8/e/38BuAb4T+lOur8YeoPupHsF0j3ZzTvo5HUvUeXZC/0/STzcu+vu55UXQtJjja4G7i3VxvJfODtwAPA14G5NdR2Pcm/4E+SzP29bVBdJHv3/yn07z3Asprr/FyoYzvJG2Z+6v7vD3XuBM6bUI1nkEy5bAe2ha/zY+vPIXXG1p8vAb4f6tkBfCC0LyT54zIF/AcwJ7QfHa5PhdsX1lznN0J/7gD+lcNH1NT2Psr7pU+oioi0UN3TMiIiUgGFu4hICyncRURaSOEuItJCCncRkRZSuIuItJDCXUSkhRTuIiIt9P+LAbNkMmemXgAAAABJRU5ErkJggg==\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "plt.imshow(grid[300:700,:].T, interpolation='nearest')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fb00ecdb-7128-42ca-9fc5-596a68ce4db7",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}