Python - 返回 dataframe 並從 function 列出

Question

我想在 Python 中創建一個 function ，它返回一個列表和一個數據框。 我知道如何使用兩個單獨的函數來做到這一點，但是有沒有辦法從單個 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)

上面的 function 只返回數據幀。 如何在同一個 function 中返回 mylist？

Answer 1

您可以簡單地返回 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)

Answer 2

只需放置逗號並添加您想要的任何內容，但請記住在 function 調用中必須給出相同的返回變量。

return new_df, my list

new_df, mylist = test_func(df)