to_dataframe.py

def to_dataframe(columns=None, iter=None):
    """
    Decorator to return a pandas DataFrame from an iterator.
    """
    def decorator(f):
        @wraps(f)
        def wrapper(*args, **kwds):
            arr = list(f(*args, **kwds))

            iter_columns = []
            if columns is not None and iter is not None:
                range_iter = range(1, len(arr[0]) - len(columns) + 1)
                iter_columns = ['{}_{}'.format(iter, i) for i in range_iter]

            return pd.DataFrame(arr, columns=columns + iter_columns)
        return wrapper
    return decorator