code: def calc_columns_index(column_name, df):
      """
      Calculate the chunk index on the axis 1 according to the selected column.
      :param column_name: selected column name
      :param df: input tiled DataFrame
      :return: chunk index on the columns axis
      """
      column_nsplits = df.nsplits[1]
      # if has duplicate columns, will return multiple values
      columns = df.columns_value.to_pandas().to_numpy()
      column_locs = (columns == column_name).nonzero()[0]
  
      return [
          np.searchsorted(np.cumsum(column_nsplits), column_loc + 1)
          for column_loc in column_locs
      ]
  
imports: import numpy
  import pandas
label_desc: Use image node_id 1 for calling a global function with example usage: calc_columns_index(column_name, df) and returns: unknown