Skip to content

Commit

Permalink
FEAT-modin-project#3303: Changed impl
Browse files Browse the repository at this point in the history
Signed-off-by: Maria Rubtsova <maria.rubtsova@intel.com>
  • Loading branch information
Rubtsowa committed Nov 23, 2021
1 parent 4cfcf46 commit 21a6fda
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions modin/pandas/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -3128,12 +3128,6 @@ def __init__(
self.__groups = self.__get_groups(*self.resample_args.values())

def __getitem__(self, key):
if len(self._dataframe.columns.intersection(key)) != len(key):
missed_keys = list(set(key).difference(self._dataframe.columns))
raise KeyError(f"Columns {str(missed_keys)[1:-1]} don't exist.")
return self._gotitem(list(key))

def _gotitem(self, key):
"""
Get define.
Expand All @@ -3147,8 +3141,12 @@ def _gotitem(self, key):
modin.pandas.BasePandasDataset
Sliced object.
"""
if len(self._dataframe.columns.intersection(key)) != len(key):
missed_keys = list(set(key).difference(self._dataframe.columns))
raise KeyError(f"Columns {str(missed_keys)[1:-1]} don't exist.")

try:
subset_ = self._dataframe[key]
subset_ = self._dataframe[list(key)]
except IndexError:
subset_ = self._dataframe

Expand Down

0 comments on commit 21a6fda

Please sign in to comment.