python函数的可选参数字符串和数据框

Question

我有执行多项任务的功能。我需要传递可选字符串/变量和可选数据框以及其他值。例如，这是我的功能。

def main(df,option=None,type=None, *args)

    if type == "cars":

       #multiple functions..

       df      = function1(df)

       results = function2(df)

       if option =="ex":

          results = function4(results)

       elif option =="CDS":

           result  = function5(results)

    elif type == "buses":

         df2 = pd_read_csv("data2",sep="\t",header=0) # This is the optional data frame I wannna pass

         def func3(df2,results,df):

             result["col3"] =pd.merge(df2,df, on="col1")

             return result

         if option =="ex":

            results = function4(results)

         elif option =="CDS":

            result  = function5(results)

     return(result)

我已经将两个可选变量传递给我的函数main，option并且type. 现在我需要再传递一个可选变量，即df2. 但我不知道该怎么做。上面的例子是我的 main 函数中的一个模板。在这里，我想添加df2并在第一个循环 formain()中使用它。elifbuses

非常感谢任何建议或帮助

Answer 1

您可以在 main 函数中使用 *args 参数来传递可选参数，例如 df2，如：def main(df,option=None,type=None, *args):    df2=args[0]    # here you will get the df2# call main function asmain(df,option,type,df2)或者您还可以使用kwargs将关键字可选参数传递为，def main(df,option=None,type=None, **kwargs):    df2=kwargs['df2']   # here you will get the df2# call main function asmain(df,option,type,df2=df2)希望这可以帮助！

Answer 2

看起来你需要。def main(df,option=None,type=None,df2=None, *args):    if df2 is not None:        #Process df2 ....