Merge pull request #7728 from dsm054/fix-get-indexerror

BUG: allow get default value upon IndexError, GH #7725

Author: jreback

doc/source/v0.15.0.txt

@@ -483,6 +483,7 @@ Enhancements
 - Bug in ``DataFrame.groupby`` where ``Grouper`` does not recognize level when frequency is specified (:issue:`7885`)
 
 
+- Bug in ``get`` where an ``IndexError`` would not cause the default value to be returned (:issue:`7725`)
 
 
 

pandas/core/generic.py

@@ -1044,7 +1044,7 @@ def get(self, key, default=None):
         """
         try:
             return self[key]
-        except (KeyError, ValueError):
+        except (KeyError, ValueError, IndexError):
             return default
 
     def __getitem__(self, item):

pandas/tests/test_generic.py

@@ -123,6 +123,23 @@ def test_get_numeric_data(self):
 
         # _get_numeric_data is includes _get_bool_data, so can't test for non-inclusion
 
+    def test_get_default(self):
+        
+        # GH 7725
+        d0 = "a", "b", "c", "d"
+        d1 = np.arange(4, dtype='int64')
+        others = "e", 10
+        
+        for data, index in ((d0, d1), (d1, d0)):
+            s = Series(data, index=index)
+            for i,d in zip(index, data):
+                self.assertEqual(s.get(i), d)
+                self.assertEqual(s.get(i, d), d)
+                self.assertEqual(s.get(i, "z"), d)
+                for other in others:
+                    self.assertEqual(s.get(other, "z"), "z")
+                    self.assertEqual(s.get(other, other), other)
+
     def test_nonzero(self):
 
         # GH 4633