Guy
Guy

Reputation: 14790

Is there a way to perform "if" in python's lambda?

In Python 2.6, I want to do:

f = lambda x: if x==2 print x else raise Exception()
f(2) #should print "2"
f(3) #should throw an exception

This clearly isn't the syntax. Is it possible to perform an if in lambda and if so how to do it?

Upvotes: 524

Views: 884815

Answers (16)

ZakS
ZakS

Reputation: 1131

it might be worth considering np.where

Upvotes: 0

Nagesh Jayaram
Nagesh Jayaram

Reputation: 27

Here's the solution if you use Python 3.x!

>>> f = lambda x: print(x) if x == 2 else print("ERROR")
>>> f(23)
ERROR
>>> f(2)
2
>>> 

Upvotes: 1

abhi
abhi

Reputation: 75

the solution for the given scenerio is:

f = lambda x : x if x == 2 else print("number is not 2")
f(30)  # number is not 2
f(2)   #2

Upvotes: 2

Anurag Bhakuni
Anurag Bhakuni

Reputation: 2439

Hope this will help a little

you can resolve this problem in the following way

f = lambda x:  x==2   

if f(3):
  print("do logic")
else:
  print("another logic")

Upvotes: 0

Paul Cysne
Paul Cysne

Reputation: 71

An easy way to perform an if in lambda is by using list comprehension.

You can't raise an exception in lambda, but this is a way in Python 3.x to do something close to your example:

f = lambda x: print(x) if x==2 else print("exception")

Another example:

return 1 if M otherwise 0

f = lambda x: 1 if x=="M" else 0

Upvotes: 2

Adarsh Somasundar
Adarsh Somasundar

Reputation: 76

This snippet should help you:

x = lambda age: 'Older' if age > 30 else 'Younger'

print(x(40))

Upvotes: 5

Manjunath Bhadrannavar
Manjunath Bhadrannavar

Reputation: 163

what you need exactly is

def fun():
    raise Exception()
f = lambda x:print x if x==2 else fun()

now call the function the way you need

f(2)
f(3)

Upvotes: 1

Rahul
Rahul

Reputation: 214

Following sample code works for me. Not sure if it directly relates to this question, but hope it helps in some other cases.

a = ''.join(map(lambda x: str(x*2) if x%2==0 else "", range(10)))

Upvotes: 0

Victor Lucas
Victor Lucas

Reputation: 58

You can also use Logical Operators to have something like a Conditional

func = lambda element: (expression and DoSomething) or DoSomethingIfExpressionIsFalse

You can see more about Logical Operators here

Upvotes: 2

Juan Pablo Lopez
Juan Pablo Lopez

Reputation: 31

If you still want to print you can import future module

from __future__ import print_function

f = lambda x: print(x) if x%2 == 0 else False

Upvotes: 3

filotn
filotn

Reputation: 566

note you can use several else...if statements in your lambda definition:

f = lambda x: 1 if x>0 else 0 if x ==0 else -1

Upvotes: 21

jimifiki
jimifiki

Reputation: 5534

Probably the worst python line I've written so far:

f = lambda x: sys.stdout.write(["2\n",][2*(x==2)-2])

If x == 2 you print,

if x != 2 you raise.

Upvotes: 44

Glenn Maynard
Glenn Maynard

Reputation: 57494

You can easily raise an exception in a lambda, if that's what you really want to do.

def Raise(exception):
    raise exception
x = lambda y: 1 if y < 2 else Raise(ValueError("invalid value"))

Is this a good idea? My instinct in general is to leave the error reporting out of lambdas; let it have a value of None and raise the error in the caller. I don't think this is inherently evil, though--I consider the "y if x else z" syntax itself worse--just make sure you're not trying to stuff too much into a lambda body.

Upvotes: 29

Robert Rossney
Robert Rossney

Reputation: 96750

The syntax you're looking for:

lambda x: True if x % 2 == 0 else False

But you can't use print or raise in a lambda.

Upvotes: 907

SilentGhost
SilentGhost

Reputation: 319651

why don't you just define a function?

def f(x):
    if x == 2:
        print(x)
    else:
        raise ValueError

there really is no justification to use lambda in this case.

Upvotes: 49

David Wolever
David Wolever

Reputation: 154504

Lambdas in Python are fairly restrictive with regard to what you're allowed to use. Specifically, you can't have any keywords (except for operators like and, not, or, etc) in their body.

So, there's no way you could use a lambda for your example (because you can't use raise), but if you're willing to concede on that… You could use:

f = lambda x: x == 2 and x or None

Upvotes: 17

Related Questions