ENH: indexing and __getitem__ of dataframe and series accept zerodim integer np.array as int

doc/source/whatsnew/v0.25.0.rst

@@ -28,6 +28,25 @@ Backwards incompatible API changes
 
 .. _whatsnew_0250.api.other:
 
+Indexing and getitem accept zerodim np.ndarray
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+
+Indexing and getitem of pd.DataFrame now accept zerodim np.array.
+
+.. ipython:: python
+    df = pd.DataFrame([[1, 2], [3, 4]])[np.array(0)]
+    df.iloc[np.array(0)]
+    df.loc[np.array(0)]
+    df[np.array(0)]
+
+Indexing and getitem of pd.Series now accept zerodim np.array.
+
+.. ipython:: python
+    sr = pd.Series([1, 2])
+    sr.iloc[np.array(0)]
+    sr.loc[np.array(0)]
+    sr[np.array(0)]
+
 Other API Changes
 ^^^^^^^^^^^^^^^^^
 

pandas/core/frame.py

@@ -2881,6 +2881,7 @@ def _ixs(self, i, axis=0):
                 return result
 
     def __getitem__(self, key):
+        key = lib.item_from_zerodim(key)
         key = com.apply_if_callable(key, self)
 
         # shortcut if the key is in columns

pandas/core/indexing.py

@@ -5,6 +5,7 @@
 import numpy as np
 
 from pandas._libs.indexing import _NDFrameIndexerBase
+from pandas._libs.lib import item_from_zerodim
 import pandas.compat as compat
 from pandas.compat import range, zip
 from pandas.errors import AbstractMethodError
@@ -1856,6 +1857,7 @@ def _getitem_axis(self, key, axis=None):
         if axis is None:
             axis = self.axis or 0
 
+        key = item_from_zerodim(key)
         if is_iterator(key):
             key = list(key)
 
@@ -2222,6 +2224,7 @@ def _getitem_axis(self, key, axis=None):
 
         # a single integer
         else:
+            key = item_from_zerodim(key)
             if not is_integer(key):
                 raise TypeError("Cannot index by location index with a "
                                 "non-integer key")

pandas/tests/indexing/test_iloc.py

@@ -675,3 +675,16 @@ def test_identity_slice_returns_new_object(self):
         # should also be a shallow copy
         original_series[:3] = [7, 8, 9]
         assert all(sliced_series[:3] == [7, 8, 9])
+
+    def test_indexing_zerodim_np_array(self):
+        # GH24919
+        df = DataFrame([[1, 2], [3, 4]])
+        result = df.iloc[np.array(0)]
+        s = pd.Series([1, 2], name=0)
+        tm.assert_series_equal(result, s)
+
+    def test_series_indexing_zerodim_np_array(self):
+        # GH24919
+        s = Series([1, 2])
+        result = s.iloc[np.array(0)]
+        assert result == 1

pandas/tests/indexing/test_loc.py

@@ -765,3 +765,16 @@ def test_loc_setitem_empty_append_raises(self):
         msg = "cannot copy sequence with size 2 to array axis with dimension 0"
         with pytest.raises(ValueError, match=msg):
             df.loc[0:2, 'x'] = data
+
+    def test_indexing_zerodim_np_array(self):
+        # GH24924
+        df = DataFrame([[1, 2], [3, 4]])
+        result = df.loc[np.array(0)]
+        s = pd.Series([1, 2], name=0)
+        tm.assert_series_equal(result, s)
+
+    def test_series_indexing_zerodim_np_array(self):
+        # GH24924
+        s = Series([1, 2])
+        result = s.loc[np.array(0)]
+        assert result == 1

pandas/tests/indexing/test_scalar.py

@@ -205,3 +205,16 @@ def test_iat_setter_incompatible_assignment(self):
         result.iat[0, 0] = None
         expected = DataFrame({"a": [None, 1], "b": [4, 5]})
         tm.assert_frame_equal(result, expected)
+
+    def test_getitem_zerodim_np_array(self):
+        # GH24924
+        # dataframe __getitem__
+        df = DataFrame([[1, 2], [3, 4]])
+        result = df[np.array(0)]
+        expected = Series([1, 3], name=0)
+        tm.assert_series_equal(result, expected)
+
+        # series __getitem__
+        s = Series([1, 2])
+        result = s[np.array(0)]
+        assert result == 1