import time import numpy as np import pandas as pd import pyarrow as pa def generate_data(total_size, ncols): nrows = int(total_size / ncols / np.dtype('float64').itemsize) return pd.DataFrame({ 'c' + str(i): np.random.randn(nrows) for i in range(ncols) })