如何基于groupby列将DataFrame按问题分组分层划分训练集与测试集

以下是各类内容的Markdown格式化规则及示例：

• 标题：使用井号 # 标识，井号数量对应标题层级（1-6级）
  示例：

  # 一级标题
  ## 二级标题
  ### 三级标题
  

• 列表项：以短横线 - 开头
  示例：

  - 第一条列表内容
  - 第二条列表内容
  

• 文本强调：用星号 * 包裹文本，单星号对应斜体，双星号对应粗体
  示例：

  *斜体强调文本*
  **粗体强调文本**
  

• 代码/命令：单行内容用反引号 ` 包裹，多行代码用三个反引号包裹，还可指定代码语言
  单行示例：

  `print("Hello World")`
  

  多行示例：

  def hello():
      print("Hello World")
  

• 引用文本：用大于号 > 开头
  示例：

  > 这是一段引用的文本内容
  

• 链接：格式为 [链接显示文本](链接地址)
  示例：

  [示例链接](https://example.com)
  

• 图片：格式为 [图片替代文本](图片地址)
  示例：

  [示例图片](https://example.com/image.jpg)
  

内容的提问来源于stack exchange，提问作者HelloWorld