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

unutbu · Accepted Answer · 2011-10-19 02:35:14Z

7

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, 2011 at 2:35

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

1 Comment

Austin Henley Over a year ago

+1 for directly answering the OP's question and enlightening me to a useful feature.

scraplesh · Accepted Answer · 2011-10-19 05:32:50Z

2

I like inline expressions, so:

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

answered Oct 19, 2011 at 5:32

scraplesh

1,41613 silver badges20 bronze badges

Comments

Austin Henley · Accepted Answer · 2011-10-19 02:38:50Z

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, 2011 at 2:38

Austin Henley

4,63313 gold badges49 silver badges80 bronze badges

1 Comment

Winston Ewert Over a year ago

I'd suggest not using flag because usually flags are boolean values.

Collectives™ on Stack Overflow

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

3 Answers 3

1 Comment

Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related