代码段如下：
import pandas as pd
train_df=pd.read_csv(r'D:\Desktop\train.csv')
test_df=pd.read_csv(r'D:\Desktop\test.csv')``
from sklearn.preprocessing import LabelEncoder
#四等分
train_df['FareBin'] = pd.qcut(train_df['Fare'],4)
#五等分
train_df['AgeBin'] = pd.cut(train_df['Age'],5)
label_encode = LabelEncoder()
labels = ['Sex','Embarked','AgeBin','FareBin']
for label in labels:
print(label,type(label))
new_label = label + '_Code'
train_df[new_label] = label_encode.fit_transform(train_df[label].astype(str))
而显示结果和提示错误为：
Sex <class 'str'>
Embarked <class 'str'>
AgeBin <class 'str'>
---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
~\Anaconda3\lib\site-packages\pandas\core\indexes\base.py in get_loc(self, key, method, tolerance)
3062 try:
-> 3063 return self._engine.get_loc(key)
3064 except KeyError:
pandas\_libs\index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas\_libs\index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas\_libs\hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
pandas\_libs\hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
KeyError: 'AgeBin'
During handling of the above exception, another exception occurred:
KeyError Traceback (most recent call last)
<ipython-input-28-619d4b1ed822> in <module>()
2 print(label,type(label))
3 new_label = label + '_Code'
----> 4 train_df[new_label] = label_encode.fit_transform(train_df[label].astype(str))
~\Anaconda3\lib\site-packages\pandas\core\frame.py in __getitem__(self, key)
2683 return self._getitem_multilevel(key)
2684 else:
-> 2685 return self._getitem_column(key)
2686
2687 def _getitem_column(self, key):
~\Anaconda3\lib\site-packages\pandas\core\frame.py in _getitem_column(self, key)
2690 # get column
2691 if self.columns.is_unique:
-> 2692 return self._get_item_cache(key)
2693
2694 # duplicate columns & possible reduce dimensionality
~\Anaconda3\lib\site-packages\pandas\core\generic.py in _get_item_cache(self, item)
2484 res = cache.get(item)
2485 if res is None:
-> 2486 values = self._data.get(item)
2487 res = self._box_item_values(item, values)
2488 cache[item] = res
~\Anaconda3\lib\site-packages\pandas\core\internals.py in get(self, item, fastpath)
4113
4114 if not isna(item):
-> 4115 loc = self.items.get_loc(item)
4116 else:
4117 indexer = np.arange(len(self.items))[isna(self.items)]
~\Anaconda3\lib\site-packages\pandas\core\indexes\base.py in get_loc(self, key, method, tolerance)
3063 return self._engine.get_loc(key)
3064 except KeyError:
-> 3065 return self._engine.get_loc(self._maybe_cast_indexer(key))
3066
3067 indexer = self.get_indexer([key], method=method, tolerance=tolerance)
pandas\_libs\index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas\_libs\index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas\_libs\hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
pandas\_libs\hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
KeyError: 'AgeBin'
请问应该如何处理？