如何将两个拆分数据组合回原始数据(numpy)

时间:2016-10-14 18:06:34

标签: python numpy

我有一个名为Total的数据,我将其拆分为功能和标签,如下所示。后来我将它们分成训练和测试数据。

Total_X = Total[:,:-1]
Total_y = Total[:,-1]

Train_X , Test_X , Train_y, Test_y = train_test_split(Total_X,Total_y,test_size = .3)

现在我想要合并Train_XTrain_y并创建一个类似Total的新列表。

1 个答案:

答案 0 :(得分:0)

你确定要先将它们分开吗?

test_len = np.floor(len(Total) * 0.3)
test_xy, train_xy = Total[:test_len], Total[test_len:]

然后,如果您需要它们,您可以提取test_x = test_xy[:,:-1]