在将我的数据集拆分回训练和测试数据集后获取空数据帧

Question

首先我用过：

train['source']='train' 
test['source']='test'
data = pd.concat([train,test],ignore_index=True)

2。执行数据探索，清理全数据，现在我想将它们分解为训练和测试数据集。

我使用了以下代码：

#Divide into test and train:
train = data.loc[data['source']=="train"]
test = data.loc[data['source']=="test"]

但是在这样做之后，训练和测试数据集是空的数据帧，只给出了列名。我做错了什么？