ageron / ageron/handson-ml

Ch2 page 58 dropping error"['income_cat'] not found in axis"

Open
#455 4 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
25.6k
Forks
12.7k
PR merge metrics
No merged PRs in 30d

Description

```python
pd.options.mode.chained_assignment = None
for set_ in (strat_train_set, strat_test_set):
set_.drop("income_cat", axis=1, inplace=True)
```

got the following error..

```
KeyError Traceback (most recent call last)
in
1 pd.options.mode.chained_assignment = None
2 for set_ in (strat_train_set, strat_test_set):
----> 3 set_.drop("income_cat", axis=1, inplace=True)

~\Anaconda3\lib\site-packages\pandas\core\frame.py in drop(self, labels, axis, index, columns, level, inplace, errors)
3695 index=index, columns=columns,
3696 level=level, inplace=inplace,
-> 3697 errors=errors)
3698
3699 @rewrite_axis_style_signature('mapper', [('copy', True),

~\Anaconda3\lib\site-packages\pandas\core\generic.py in drop(self, labels, axis, index, columns, level, inplace, errors)
3109 for axis, labels in axes.items():
3110 if labels is not None:
-> 3111 obj = obj._drop_axis(labels, axis, level=level, errors=errors)
3112
3113 if inplace:

~\Anaconda3\lib\site-packages\pandas\core\generic.py in _drop_axis(self, labels, axis, level, errors)
3141 new_axis = axis.drop(labels, level=level, errors=errors)
3142 else:
-> 3143 new_axis = axis.drop(labels, errors=errors)
3144 result = self.reindex(**{axis_name: new_axis})
3145

~\Anaconda3\lib\site-packages\pandas\core\indexes\base.py in drop(self, labels, errors)
4402 if errors != 'ignore':
4403 raise KeyError(
-> 4404 '{} not found in axis'.format(labels[mask]))
4405 indexer = indexer[~mask]
4406 return self.delete(indexer)

KeyError: "['income_cat'] not found in axis"
```

I checked both the sets with `head()` method and I have the `income_cat` column in them but still I am getting the error. please help.!!

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.