首页 > 解决方案 > 在数据框中拆分行并保存到新数据集中

问题描述

问题是:将 clean_mlb 的行拆分为由非匹兹堡球队的 29 行组成的训练集和对应于匹兹堡海盗队的一个观察值的测试集。将这些数据帧保存为训练集和测试集。

clean_mlb 是由名为 mlb 的数据集组成的数据帧

数据集 MLB

数据框 clean_mlb

这是我的代码:

  if(mlb$TM[i] == "PIT"){
    
  }else{

  }
} 

我不知道在 if 或 else 之后应该放什么,或者这甚至是我应该做的。

标签: rdataframe

解决方案


您可以subset根据列中的值获取数据TM

train <- subset(mlb, TM != "PIT")
test <- subset(mlb, TM == "PIT")

推荐阅读