Python：为函数中的函数设置参数

Question

我有一个函数，它具有多个函数，所有这些函数都带有2个必需参数和许多可选参数。我想知道如何在这个函数中设置可选的参数给定函数：

chart_type = "bar" 

def chart_selector(slide, df_table, chart_type): 

    if chart_type == "bar": 
     add_bar_chrt(slide, df_table) 
    elif chart_type == "column": 
     add_column_chrt(slide, df_table) 
    elif chart_type == "pie": 
     add_pie_chrt(slide, df_table) 
    elif chart_type == "line": 
     add_line_chrt(slide, df_table) 

这里是我想怎么办：我想使用chart_selector()功能，如果chart_type是"bar"，然后我想要为add_bar_chrt()函数设置几个可选参数，但我不知道如何？

，所以我需要它以某种方式加入到这个代码：

chart = chart_selector(slide, df_table, chart_type) 

Answer 1

您可以通过使用*args和**kwargs给你的函数签名添加任意的论点支持，再传给那些：

def chart_selector(slide, df_table, chart_type, *args, **kwargs): 
    if chart_type == "bar": 
     add_bar_chrt(slide, df_table, *args, **kwargs) 

任何额外您现在传递给chart_selector()的论点现在转交给add_bar_chrt()。

当你在这个函数的工作，无论如何，考虑使用字典派遣图表类型：

chart_types = { 
    'bar': add_bar_chrt, 
    'column': add_column_chrt, 
    'pie': add_pie_chart, 
    'line': add_line_chart, 
} 
def chart_selector(slide, df_table, chart_type, *args, **kwargs): 
    return chart_types[chart_type](slide, df_table, *args, **kwargs) 

字典取代多分枝if .. elif ..结构。

演示：

>>> def add_bar_chrt(slide, tbl, size=10, color='pink'): 
...  return 'Created a {} barchart, with bars size {}'.format(size, color) 
... 
>>> def add_column_chrt(slide, tbl, style='corinthyan', material='marble'): 
...  return 'Created a {} column chart, with {}-style plinths'.format(material, style) 
... 
>>> chart_types = { 
...  'bar': add_bar_chrt, 
...  'column': add_column_chrt, 
... } 
>>> def chart_selector(slide, df_table, chart_type, *args, **kwargs): 
...  return chart_types[chart_type](slide, df_table, *args, **kwargs) 
... 
>>> chart_selector('spam', 'eggs', 'bar') 
'Created a 10 barchart, with bars size pink' 
>>> chart_selector('spam', 'eggs', 'column', material='gold') 
'Created a gold column chart, with corinthyan-style plinths'