Return two data frames from a function with data f

2020-04-05 08:25发布

问题:

I want to return two data frames from a function, like this:

def test():
    df1 = pd.DataFrame([1,2,3], ['a','b','c'])
    df2 = pd.DataFrame([4,5,6], ['d','e','f'])
    return df1
    return df2
test()

But the function only returns one data frame df1. How to return both in pretty data frame format, not in cmd black background format?

When I tried to return both using

return df1, df2

in the Jupyter Notebook, the output returns the data frames in a black background cmd-like format, and not in proper data frame format.

回答1:

How about this:

def test():
    df1 = pd.DataFrame([1,2,3], ['a','b','c'])
    df2 = pd.DataFrame([4,5,6], ['d','e','f'])
    return df1, df2

a, b = test()
display(a, b)

This prints out:

    0
a   1
b   2
c   3

    0
d   4
e   5
f   6