如何將兩個拆分數據合併爲原始（numpy）

-2

我有一個名爲Total的數據，我將它分解爲如下所示的要素和標籤。後來我將它們分成訓練和測試數據。如何將兩個拆分數據合併爲原始（numpy）

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。

'np.column_stack（（Train_X，Train_Y））'？ – Divakar

任何重組您都會得到一份副本。你可以顯示train_test_split的實現嗎？ – Eric

你確定你想把他們分開嗎？

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

然後，如果需要的話，你可以提取test_x = test_xy[:,:-1]等

2016-10-15 00:30:19 Eric

回答