Source code for neurokit2.epochs.epochs_to_df

import pandas as pd




def epochs_to_df(epochs):
    """**Convert epochs to a DataFrame**

    Convert epochs to a DataFrame.

    Parameters
    ----------
    epochs : dict
        A dict containing one DataFrame per event/trial. Usually obtained via `epochs_create()`.


    Returns
    ----------
    DataFrame
        A DataFrame containing all epochs identifiable by the 'Label' column, which time axis
        is stored in the 'Time' column.

    See Also
    ----------
    events_find, events_plot, epochs_create, epochs_plot

    Examples
    ----------
    .. ipython:: python

      import neurokit2 as nk

      # Get data
      data = nk.data("bio_eventrelated_100hz")

      # Find events
      events = nk.events_find(data["Photosensor"],
                              threshold_keep='below',
                              event_conditions=["Negative", "Neutral", "Neutral", "Negative"])

      # Create epochs
      epochs = nk.epochs_create(data, events, sampling_rate=200, epochs_end=3)

      # Convert to DataFrame
      data = nk.epochs_to_df(epochs)
      data.head()

    """
    data = pd.concat(epochs)
    data["Time"] = data.index.get_level_values(1).values
    data = data.reset_index(drop=True)

    return data



def _df_to_epochs(data):
    # Convert dataframe of epochs created by `epochs_to_df` back into a dictionary.
    return {label: group.set_index("Time") for label, group in data.groupby("Label")}