我有一个名为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_X
和Train_y
并创建一个类似Total
的新列表。
答案 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]
等