Pandas dataframe - 將 N 行從一個 dataframe 移動到另一個

Question

我有一個用於機器學習的訓練集和測試集，但是訓練集包含的數據行太多，而測試集包含的數據太少。 我計算出我需要將 245 行從訓練集移動到測試集以產生更好的分割。 我怎樣才能做到這一點？ 我在訓練集中總共有 5116 行。

首先，我使用這個隨機化了訓練集的行

train_df = train_df.sample(n = len(train_df)).reset_index(drop=True)

然后我想抓取最后 245 行並將它們移動到test_df

我在這里找到了這兩個解決方案

Pandas dataframe - 將行從一個 dataframe 移動到另一個

和

Pandas 將行從 1 個 DF 移動到另一個 DF

但是，他們正在根據我沒有的條件選擇行。 如果可能的話，我有點想像在 python 中使用 arrays 上的 slice 那樣做。

也許像（第 0-5116 - 245 行和從 0 開始的所有列）

transferdata_df = train_df.iloc[5115 - 245:, 0:]

然后 append 到測試集就好了

test_df.append(transferdata_df)

我不確定這是否是正確的方法。

Answer 1

讓我們做

transferdata_df = train_df.iloc[- 245:, 0:]

test_df = test_df.append(transferdata_df)

train_df =train_df.drop(transferdata_df.index)