How to pass an operator to a python function?

Question

I'd like to pass a math operator, along with the numeric values to compare, to a function. Here is my broken code:

def get_truth(inp,relate,cut):    
    if inp print(relate) cut:
        return True
    else:
        return False

and call it with

get_truth(1.0,'>',0.0)

which should return True.

grc · Accepted Answer · 2013-09-03T12:06:44.883

91

Have a look at the operator module:

import operator
get_truth(1.0, operator.gt, 0.0)

...

def get_truth(inp, relate, cut):    
    return relate(inp, cut)
    # you don't actually need an if statement here

edited Sep 03 '13 at 12:06

answered Sep 03 '13 at 12:01

grc

21,645
4
39
63

In the end, I just got rid of my function and used operator. 'if operator.gt(1.0,0.0):' – philshem Sep 03 '13 at 12:43
Couldn't you just use `if 1.0 > 0.0:`? – grc Sep 03 '13 at 13:38
2

I want to be able to change the operator dynamically (so I may have to go back to the function solution). – philshem Sep 03 '13 at 15:28
1

for `in` operator see `operator.contains(a,b)`. Note that this checks presence of b in a not the other way round. – Ilia Gilmijarow Sep 25 '18 at 14:28

score 51 · Answer 2 · edited Jun 15 '21 at 05:12

51

Make a mapping of strings and operator functions. Also, you don't need if/else condition:

import operator


def get_truth(inp, relate, cut):
    ops = {'>': operator.gt,
           '<': operator.lt,
           '>=': operator.ge,
           '<=': operator.le,
           '==': operator.eq}
    return ops[relate](inp, cut)


print(get_truth(1.0, '>', 0.0)) # prints True
print(get_truth(1.0, '<', 0.0)) # prints False
print(get_truth(1.0, '>=', 0.0)) # prints True
print(get_truth(1.0, '<=', 0.0)) # prints False
print(get_truth(1.0, '==', 0.0)) # prints False

FYI, eval() is evil: Why is using 'eval' a bad practice?

edited Jun 15 '21 at 05:12

Michael Dorner

14,449
11
73
105

answered Sep 03 '13 at 12:02

alecxe

441,113
110
1,021
1,148

3

+1. Also, this answer demonstrates that you can directly return the value of `ops[relate]`, instead of explicitly testing it and returning a literal `True` or `False`. – chepner Sep 03 '13 at 12:05
1

I recommend to replace `=` with `==` – leoll2 Mar 12 '21 at 15:39
Beautiful! But please use `==` for comparison! – Michael Dorner Jun 14 '21 at 06:58

Viktor Kerkez · Answer 3 · 2013-09-03T12:07:32.033

12

Use the operator module. It contains all the standard operators that you can use in python. Then use the operator as a functions:

import operator

def get_truth(inp, op, cut):
    return op(inp, cut):

get_truth(1.0, operator.gt, 0.0)

If you really want to use strings as operators, then create a dictionary mapping from string to operator function as @alecxe suggested.

edited Sep 03 '13 at 12:07

answered Sep 03 '13 at 12:01

Viktor Kerkez

41,890
11
100
83

score -1 · Answer 4 · edited May 10 '20 at 09:04

-1

Use the operator module instead:

import operator
def get_truth(inp, relate, cut):
    rel_ops = {
        '>': operator.gt,
        '<': operator.lt,
        '>=': operator.ge,
        '<=': operator.le,
        '==': operator.eq,
        '!=': operator.ne
    }
    return rel_ops[relate](inp, cut)

edited May 10 '20 at 09:04

E_net4 - Krabbe mit Hüten

24,143
12
85
121

answered Sep 03 '13 at 12:01

Paul Evans

26,798
3
34
52

@PaulEvans: because mapping ` – Martijn Pieters Sep 03 '13 at 12:10
Thanks @PaulEvans. I will go with the operator method for reasons desribed here: http://stackoverflow.com/questions/1832940/is-using-eval-in-python-a-bad-practice – philshem Sep 03 '13 at 12:13

score -3 · Answer 5 · answered Sep 03 '13 at 12:07

-3

>>> def get_truth(inp,relate,cut):
...     if eval("%s%s%s" % (inp,relate,cut)):
...         return True
...     else:
...         return False
...
>>> get_truth(1.0,'>',0.0)
True
>>>

answered Sep 03 '13 at 12:07

amadain

2,516
3
36
52

11

No, please don't use `eval()` when better solutions are available. – Martijn Pieters Sep 03 '13 at 12:11
what is your problem with eval??? You are downvoting everybody who has given a working solution that uses eval – amadain Sep 03 '13 at 12:13
8

`eval()` poses a security risk, and is way overused as a 'solution' whenever people do not know a better option exists. – Martijn Pieters Sep 03 '13 at 12:15

How to pass an operator to a python function?

5 Answers5

Linked

Related