code: def generate_chunks(
      total_size, nchunks, ncols, dtype=np.dtype("int64")
  ):
      rowsize = total_size // nchunks // ncols
      assert rowsize % dtype.itemsize == 0
  
      def make_column(col, chunk):
          return np.frombuffer(
              common.get_random_bytes(rowsize, seed=col + 997 * chunk)
          ).view(dtype)
  
      return [
          pd.DataFrame(
              {
                  "c" + str(col): make_column(col, chunk)
                  for col in range(ncols)
              }
          )
          for chunk in range(nchunks)
      ]
  
file_path: arrow/python/benchmarks/streaming.py
imports: import numpy
  import pandas
  import pyarrow
  from .None import common
  from .common import KILOBYTE, MEGABYTE
label_desc: Use image node_id 1 for calling a global function with example usage: generate_chunks(total_size, nchunks, ncols, dtype) and returns: unknown, np