code: def select_data_from_prediction_picking(
      round_dataframe: pd.DataFrame, selected_data: dict
  ) -> pd.DataFrame:
      """Create a subset dataframe from the prediction picking graph selection.
  
      Parameters
      ----------
      round_dataframe : pd.DataFrame
          Data to sample
      selected_data : dict
          Selected sample in the prediction picking graph
  
      Returns
      -------
      pd.DataFrame
          Subset dataframe
      """
      row_ids = []
      for p in selected_data["points"]:
          row_ids.append(p["customdata"])
      df = round_dataframe.loc[row_ids]
  
      return df
  
imports: from typing import Optional, Tuple
  import numpy
  import pandas
  import datetime
  from dash import dcc, html
  import dash_bootstrap_components
label_desc: Use image node_id 1 for calling a global function with example usage: select_data_from_prediction_picking(round_dataframe, selected_data) and returns: df