tools.py

import pandas as pd

def make_dataframe(raw_dat):
    df = pd.DataFrame(
        dict(
            time=raw_dat.t,
            type_1=raw_dat.y[0, :],
            type_2=raw_dat.y[1, :],
            type_3=raw_dat.y[2, :],
            total=raw_dat.y[3, :],
        ))
    return df