Python等价于r中的子集函数

Question

我根本不知道python，但我目前正在进行的项目必须使用它来完成，我有这个r代码

y_train <- subset(train_df_cleaned_2, Id==IdExec)$Y

我需要在python中做完全相同的事情。我该怎么做？谢谢

Answer 1

Python中实现这一点的标准方法是理解，例如：

y_train = [x for x in train_df_cleaned_2 if x.Id == IdExec]

Answer 2

我用这个找到了这个答案：子设置Python DataFrame

y = train[(train.NflId == train.NflIdRusher)][['Yards']]

感谢大家的帮助