hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cast locator with index type (#2274)

Browse source
This commit is contained in:
Simon Mo 2018-06-21 15:28:22 +00:00 • committed by Devin Petersohn
parent aa5daa1b82
commit b108419628
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/ray/dataframe/indexing.py
View file

@ -370,7 +370,11 @@ class _Loc_Indexer(_Location_Indexer_Base):
Returns:
nan_labels: The labels needs to be added
"""
locator_as_index = pandas.Index(locator)
# base_index_type can be pd.Index or pd.DatetimeIndex
# depending on user input and pandas behavior
# See issue #2264
base_index_type = type(base_index)
locator_as_index = base_index_type(locator)
nan_labels = locator_as_index.difference(base_index)
common_labels = locator_as_index.intersection(base_index)