CODEWITHSUNDEEP
pythoncoding-style
dgamma3
dgamma3

Reputation: 2371

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

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

Upvotes: 1

Views: 752

Answers (3)

scraplesh
scraplesh

Reputation: 1386

I like inline expressions, so:

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

Upvotes: 2

Austin Henley
Austin Henley

Reputation: 4633

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.

Upvotes: 1

unutbu
unutbu

Reputation: 880547

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)

Upvotes: 7

Related Questions