Calling a certain function based on a variable

Question

I am trying to call certain functions based on a certain variable. If I have lots of functions based on this variable it quickly gets very ugly with all if statements. My questions is if there is a more elegant and "pythonic" solution than doing something like the code below?

if variable == 0:
    function_0()
elif variable == 1:
    function_1()
elif variable == 2:
    function_2()

Is the test always for equality? Do you need to test something like ``variable > 5``, for example? — MisterMiyagi, Oct 12 '20 at 14:43

score 4 · Accepted Answer · answered Oct 12 '20 at 14:43

4

Create an array of the functions, index with the variable and call the function.

[function_0, function_1, function_2][variable]()

Or do it via a dictionary

dd = {0 : function_0, 1 : function_1, 2 : function_2}
dd[variable]()

answered Oct 12 '20 at 14:43

rioV8

24,506
3
32
49

score 0 · Answer 2 · answered Oct 12 '20 at 14:44

0

You could use a dictionary

fmap = {0: f1, 1: f2, 2: f3}
fmap[variable]()

answered Oct 12 '20 at 14:44

OneCricketeer

179,855
19
132
245

Calling a certain function based on a variable

2 Answers2