Multidimensional array to pandas dataframe

array_to_dataframe.py

import pandas as pd
from typing import Optional, List


def array_to_dataframe(array, axis_names: Optional[List[str]]=None):
    """Based on https://stackoverflow.com/questions/35525028/how-to-transform-a-3d-arrays-into-a-dataframe-in-python"""
    if axis_names is None:
        axis_names = list(range(array.ndim))
    index = pd.MultiIndex.from_product([range(s) for s in array.shape], names=names)
    df = pd.DataFrame({"array": array.flatten()}, index=index)["array"]
    return df
  
if __name__ == "__main__":
    import numpy as np
    x = np.random.randn((2,3,4))
    array_to_dataframe(x)