Python - return dataframe and list from function

Question

I want to create a function in Python that returns both a list and a data frame. I know how to do this with two seperate functions, but is there a way to return both from a single function?

import pandas as pd
# sample data
data_dict = {'unit':['a','b','c','d'],'salary':[100,200,250,300]}
# create data frame
df = pd.DataFrame(data_dict)
# Function that returns a dataframe
def test_func(df):
    # create list meeting some criteria
    mylist = list(df.loc[(df['salary']<=200),'unit'])
    # create new dataframe based on mylist
    new_df = df[df['unit'].isin(mylist)]
    return new_df
# create dataframe from function
new_df = test_func(df)

The above function returns just the data frame. How do I also return mylist in the same function?

Does this answer your question? How do I return multiple values from a function? — Daniel Geffen
– Daniel Geffen, Commented Apr 28, 2020 at 15:12

Keyb0ardwarri0r · Accepted Answer · 2020-04-28 15:12:01Z

4

You can simply return two variables in the function

import pandas as pd
# sample data
data_dict = {'unit':['a','b','c','d'],'salary':[100,200,250,300]}
# create data frame
df = pd.DataFrame(data_dict)
# Function that returns a dataframe
def test_func(df):
    # create list meeting some criteria
    mylist = list(df.loc[(df['salary']<=200),'unit'])
    # create new dataframe based on mylist
    new_df = df[df['unit'].isin(mylist)]
    return new_df, my list

# create dataframe and list from function
new_df, mylist = test_func(df)

answered Apr 28, 2020 at 15:12

Keyb0ardwarri0r

2373 silver badges12 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Ali Hassan · Accepted Answer · 2020-04-28 15:22:13Z

0

just place comma and add anything you want, but remember in function calling must give same return variables.

return new_df, my list

new_df, mylist = test_func(df)

answered Apr 28, 2020 at 15:22

Ali Hassan

96611 silver badges25 bronze badges

Collectives™ on Stack Overflow

Python - return dataframe and list from function

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related