python: make a variable equal an operator (+,/,*,-)

Question

is it possible to assign a variable a maths operator.

this is what I've currently got, just a sample (typed it in now, so dont worry about simple errors)

if image == "lighten":
    red_channel = red_channel + 50
else:  // image is darken
    red_channel = red_channel  - 50

notice how i am repeating the exact same code, with a different operator. Is it possible to achieve something like this:

if (image == "lighten"):
    operator = +
else:
    operator = - 

red_channel = red_channel operator 50

score 6 · Accepted Answer · answered Oct 19 '11 at 02:35

6

import operator
if (image == 'lighten'):
    op = operator.add
else:
    op = operator.sub

red_channel = op(red_channel, 50)

Or, if you have a number of possible operations,

op = {
    'lighten':operator.add,
    'darken':operator.sub,
     ...
    }
red_channel = op[image](red_channel,50)

answered Oct 19 '11 at 02:35

unutbu

842,883
184
1,785
1,677

1

+1 for directly answering the OP's question and enlightening me to a useful feature. – Austin Henley Oct 19 '11 at 02:41

score 2 · Answer 2 · answered Oct 19 '11 at 05:32

2

I like inline expressions, so:

red_channel += 50 if image == 'lighten' else -50

answered Oct 19 '11 at 05:32

scraplesh

1,326
12
20

score 1 · Answer 3 · answered Oct 19 '11 at 02:38

1

Another option rather than going to that length, as long as you are only doing either positive 50 or negative 50 is:

red_channel = red_channel + (flag * 50)

The variable "flag" is either 1 or -1; thus giving you 50 or -50. This won't save a lot of code for this small example, but I use it at times when it is convenient.

answered Oct 19 '11 at 02:38

Austin Henley

4,625
13
45
80

I'd suggest not using `flag` because usually flags are boolean values. – Winston Ewert Oct 19 '11 at 03:35

python: make a variable equal an operator (+,/,*,-)

3 Answers3

Linked