将数据帧分为列条件的训练和测试数据集

Question

我尝试使用train_test_split病态学习函数创建X_train，X_test，Y_train和Y_test数据集，但我想添加一个条件，即date_birth列中的所有值必须在19 / 01/01和19/06/01。

您能帮我在下面的代码中添加此条件吗？

谢谢

# Create the train and test data
X_train, X_test, y_train, y_test = train_test_split(card_df.drop(["state"], axis=1),
                                                    card_df.state,
                                                    stratify=card_df.state,
                                                    test_size=0.3,
                                                    random_state=1984)