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

doc/source/whatsnew/v0.25.0.rst

@@ -19,6 +19,7 @@ including other versions of pandas.
 Other Enhancements
 ^^^^^^^^^^^^^^^^^^
 
+- Indexing and __getitem__ of DataFrame and Series now accept zerodim np.ndarray. (:issue:`24919`)
 - :meth:`Timestamp.replace` now supports the ``fold`` argument to disambiguate DST transition times (:issue:`25017`)
 - :meth:`DataFrame.at_time` and :meth:`Series.at_time` now support :meth:`datetime.time` objects with timezones (:issue:`24043`)
 -

pandas/core/frame.py

@@ -2838,6 +2838,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

@@ -697,3 +697,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

@@ -778,3 +778,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

@@ -221,3 +221,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