Source code for utility_modules.math_functions

#!/usr/bin/env python
def expand_grid(*args, **kwargs):
    """
    functionally equivlant as R expand.grid fucntion
    http://stackoverflow.com/questions/12130883/r-expand-grid-function-in-python
    """
    
    import itertools as it
    import numpy as np
    import pandas as pd

    columns = []
    lst = []
    if args:
        columns += xrange(len(args))
        lst += args
    if kwargs:
        columns += kwargs.iterkeys()
        lst += kwargs.itervalues()
    return pd.DataFrame(list(it.product(*lst)), columns=columns)