数据挖掘 - 熊猫索引错误 - 吾爱随笔录

我正在尝试使用train_test_split拆分我的数据。但是，我收到一个索引错误。我在下面粘贴了部分错误消息。我正在使用 Python 3.5 版本和sklearn0.18.1。该代码适用于我以前不同的数据集。这里的特征在 Pandas 中DataFrame，标签在 Pandas 中 Series。

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(features, labels, test_size=0.4, random_state=1)

KeyError                                  
Traceback (most recent call last)
/apps/anaconda/anaconda-3.5/lib/python3.5/site-
    packages/pandas/indexes/base.py in get_loc(self, key, method, tolerance)
   2133             try:
-> 2134                 return self._engine.get_loc(key)
   2135             except KeyError:<br><br>
pandas/index.pyx in pandas.index.IndexEngine.get_loc (pandas/index.c:4443)()<br><br>
pandas/index.pyx in pandas.index.IndexEngine.get_loc (pandas/index.c:4289)()<br><br>
pandas/src/hashtable_class_helper.pxi in pandas.hashtable.PyObjectHashTable.get_item (pandas/hashtable.c:13733)()<br><br>
pandas/src/hashtable_class_helper.pxi in pandas.hashtable.PyObjectHashTable.get_item (pandas/hashtable.c:13687)()<br><br>
KeyError: 0