code: def _select_dataframe_columns(df, cols):
      if isinstance(cols[0], str):
          inputs = df.loc[:, cols].values
      elif isinstance(cols[0], int):
          inputs = df.iloc[:, cols].values
      else:
          raise ValueError(
              "Provided columns should be string column names or integer column indices"
          )
      return inputs
  
file_path: pywick/pywick/datasets/CSVDataset.py
imports: from .BaseDataset import BaseDataset
  import numpy
  import pandas
  from .data_utils import _return_first_element_of_list, default_file_reader, _pass_through, _process_transform_argument, _process_co_transform_argument
label_desc: Use image node_id 3 for calling a global function with example usage: _select_dataframe_columns(df, cols) and returns: inputs