Byte89 get_state_abbrev

Author: ushills

days/07-09-data-structures/code/Byte89.ipynb

@@ -47,15 +47,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 25,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "30.5 µs ± 478 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)\n",
-      "['Massachusetts', 'Missouri', 'Hawaii', 'Vermont', 'Delaware']\n"
+      "30.4 µs ± 563 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)\n",
+      "['Oregon', 'Massachusetts', 'Nebraska', 'Missouri', 'Alaska', 'Hawaii', 'Maryland', 'Vermont', 'South Dakota', 'Delaware']\n"
      ]
     }
    ],
@@ -72,7 +72,33 @@
     "    return(listofnthstates)      \n",
     "\n",
     "%timeit get_every_nth_state(10)\n",
-    "print(get_every_nth_state(10))"
+    "print(get_every_nth_state(5))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 26,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "295 ns ± 17.2 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)\n",
+      "WI\n"
+     ]
+    }
+   ],
+   "source": [
+    "def get_state_abbrev(abbrev):\n",
+    "    try:\n",
+    "        state_abbrev = us_state_abbrev[abbrev]\n",
+    "        return state_abbrev\n",
+    "    except KeyError:\n",
+    "        return NOT_FOUND\n",
+    "    \n",
+    "%timeit get_state_abbrev('Wisconsin')\n",
+    "print(get_state_abbrev('Wisconsin'))"
    ]
   },
   {

days/07-09-data-structures/code/states.py

@@ -44,12 +44,12 @@ def get_every_nth_state(n=10):
     return(listofnthstates)
 
 
-
 def get_state_abbrev(abbrev):
-    """Look up a state abbreviation by full name in
-       us_state_abbrev, if not found return the string stored in the
-       NOT_FOUND constant (takeaway: dicts are great for lookups)"""
-    pass
+    try:
+        state_abbrev = us_state_abbrev[abbrev]
+        return state_abbrev
+    except KeyError:
+        return NOT_FOUND
 
 
 def get_longest_state(data):
@@ -65,5 +65,6 @@ def combine_state_names_and_abbreviations():
        values and use sorted, list slicing and list concatenation)"""
     pass
 
+
 if __name__ == '__main__':
     print(get_every_nth_state(10))