BUG: iloc raising for ea series #50171
Conversation
phofl
added
Indexing
There was a problem hiding this comment.
See also my more general comment about using IntegerArrays in .iloc in #49521.
| df = DataFrame([[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]) | ||
| result = df.iloc[Series([1], dtype="Int64"), Series([0, 1], dtype="Int64")] | ||
| expected = DataFrame([[5, 6]], index=[1]) | ||
| tm.assert_frame_equal(result, expected) |
There was a problem hiding this comment.
I'm pretty suire this would fail if a pd.NA was in the indexers?
There was a problem hiding this comment.
Yep it does, which is fine. Will add a test
| elif is_list_like_indexer(key): | ||
| arr = np.array(key) | ||
| if isinstance(key, ABCSeries): | ||
| arr = key._values |
There was a problem hiding this comment.
How about integer arrays with pd.NA?
There was a problem hiding this comment.
The errors are different for Series and IntegerArray:
>>> import pandas as pd
>>> df = pd.DataFrame([[0,1,2,3,4],[5,6,7,8,9]])
>>> iarr = pd.array([0,1,2, pd.NA], dtype = pd.Int64Dtype())
>>> df.iloc[:, iarr]
IndexError: .iloc requires numeric indexers, got [0 1 2 <NA>]
df.iloc[:, pd.Series(iarr)]
ValueError: cannot convert to 'int64'-dtype NumPy array with missing values. Specify an appropriate 'na_value' for this dtype.I think the ValueError should be raised in both cases?
There was a problem hiding this comment.
I guess you tried the series case on main? I am getting consistent errors on my pr.
In General I would prefer a better suited error, pointing to NA in iloc. But will probably do in a follow up. We will have to check this later, because not all cases get here and this does not cover the series.iloc case
There was a problem hiding this comment.
Sorry wrong snippet. It's the non-NA array:
>>> import pandas as pd
>>> df = pd.DataFrame([[0,1,2,3,4],[5,6,7,8,9]])
>>> iarr = pd.array([0,1,2], dtype = pd.Int64Dtype())
>>> df.iloc[:, iarr]
IndexError: .iloc requires numeric indexers, got [0 1]There was a problem hiding this comment.
Ah got you, fixed that as well
|
Thanks @phofl |
3 participants
doc/source/whatsnew/vX.X.X.rstfile if fixing a bug or adding a new feature.