Python: defining my own operators?

I would like to define my own operator. Does python support such a thing?

Answers:

Thank you for visiting the Q&A section on Magenaut. Please note that all the answers may not help you solve the issue immediately. So please treat them as advisements. If you found the post helpful (or not), leave a comment & I’ll get back to you as soon as possible.

Method 1

While technically you cannot define new operators in Python, this clever hack works around this limitation. It allows you to define infix operators like this:

# simple multiplication
x=Infix(lambda x,y: x*y)
print 2 |x| 4
# => 8

# class checking
isa=Infix(lambda x,y: x.__class__==y.__class__)
print [1,2,3] |isa| []
print [1,2,3] <<isa>> []
# => True

Method 2

No, Python comes with a predefined, yet overridable, set of operators.

Method 3

No, you can’t create new operators. However, if you are just evaluating expressions, you could process the string yourself and calculate the results of the new operators.

Method 4

Sage provides this functionality, essentially using the “clever hack” described by @Ayman Hourieh, but incorporated into a module as a decorator to give a cleaner appearance and additional functionality – you can choose the operator to overload and therefore the order of evaluation.

from sage.misc.decorators import infix_operator

@infix_operator('multiply')
def dot(a,b):
    return a.dot_product(b)
u=vector([1,2,3])
v=vector([5,4,3])
print(u *dot* v)
# => 22

@infix_operator('or')
def plus(x,y):
    return x*y
print(2 |plus| 4)
# => 6

See the Sage documentation and this enhancement tracking ticket for more information.

Method 5

Python 3.5 introduces the symbol @ for an extra operator.

PEP465 introduced this new operator for matrix multiplication, to simplify the notation of many numerical code. The operator will not be implemented for all types, but just for arrays-like-objects.

You can support the operator for your classes/objects by implementing __matmul__().

The PEP leaves space for a different usage of the operator for non-arrays-like objects.

Of course you can implement with @ any sort of operation different from matrix multiplication also for arrays-like objects, but the user experience will be affected, because everybody will expect your data type to behave in a different way.

Method 6

If you intend to apply the operation on a particular class of objects, you could just override the operator that matches your function the closest… for instance, overriding __eq__() will override the == operator to return whatever you want. This works for almost all the operators.

All methods was sourced from stackoverflow.com or stackexchange.com, is licensed under cc by-sa 2.5, cc by-sa 3.0 and cc by-sa 4.0