chosing a function without if statement in python

Question

Imagine having 200 functions, which represent 200 ways to solve a problem or calculate something like

def A():
   ...
   
def B():
   ...
.
.
.

and the method will be chosen as an input argument, meaning the user decides which method to use, giving this as an argument while running the program like "A" for function/method A. how to chose that function without if-checking the name of every single function in python.

You could put your functions in a dictionary, or access them via the globals() dictionary. — khelwood
– khelwood, Commented Jul 30, 2020 at 12:25

Mohamed Ali JAMAOUI · Accepted Answer · 2020-07-30 12:31:18Z

3

You can use a dictionary to access directly the function that you need in O(1) complexity. For example:

def A(x):
   pass

def B(x):
   pass

func_map = {"A": A, "B": B}

Say that you store the user input in a variable chosen_func, then to select and run the right function, do the following:

func_map[chosen_func](x)

Example:

In [1]: def A(x): 
   ...:     return x + x 
 
In [2]: def B(x): 
   ...:     return x * x  

In [3]: func_map = {"A": A, "B": B}

In [4]: func_map["A"](10)
Out[4]: 20

In [5]: func_map["B"](10)
Out[5]: 100

edited Jul 30, 2020 at 12:31

answered Jul 30, 2020 at 12:24

Mohamed Ali JAMAOUI

14.8k14 gold badges79 silver badges124 bronze badges

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

Collectives™ on Stack Overflow

chosing a function without if statement in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related