python if else with assignment

Python Conditional Assignment

When you want to assign a value to a variable based on some condition, like if the condition is true then assign a value to the variable, else assign some other value to the variable, then you can use the conditional assignment operator.

In this tutorial, we will look at different ways to assign values to a variable based on some condition.

1. Using Ternary Operator

The ternary operator is very special operator in Python, it is used to assign a value to a variable based on some condition.

It goes like this:

Here, the value of variable will be value_if_true if the condition is true, else it will be value_if_false .

Let's see a code snippet to understand it better.

You can see we have conditionally assigned a value to variable c based on the condition a > b .

2. Using if-else statement

if-else statements are the core part of any programming language, they are used to execute a block of code based on some condition.

Using an if-else statement, we can assign a value to a variable based on the condition we provide.

Here is an example of replacing the above code snippet with the if-else statement.

3. Using Logical Short Circuit Evaluation

Logical short circuit evaluation is another way using which you can assign a value to a variable conditionally.

The format of logical short circuit evaluation is:

It looks similar to ternary operator, but it is not. Here the condition and value_if_true performs logical AND operation, if both are true then the value of variable will be value_if_true , or else it will be value_if_false .

Let's see an example:

But if we make condition True but value_if_true False (or 0 or None), then the value of variable will be value_if_false .

So, you can see that the value of c is 20 even though the condition a < b is True .

So, you should be careful while using logical short circuit evaluation.

While working with lists , we often need to check if a list is empty or not, and if it is empty then we need to assign some default value to it.

Let's see how we can do it using conditional assignment.

Here, we have assigned a default value to my_list if it is empty.

Assign a value to a variable conditionally based on the presence of an element in a list.

Now you know 3 different ways to assign a value to a variable conditionally. Any of these methods can be used to assign a value when there is a condition.

The cleanest and fastest way to conditional value assignment is the ternary operator .

if-else statement is recommended to use when you have to execute a block of code based on some condition.

Happy coding! 😊

Learn Python practically and Get Certified .

Popular Tutorials

Popular examples, reference materials, learn python interactively, python introduction.

• Getting Started
• Keywords and Identifier
• Python Comments
• Python Variables
• Python Data Types
• Python Type Conversion
• Python I/O and Import
• Python Operators
• Python Namespace

Python Flow Control

• Python if...else
• Python for Loop

Python while Loop

Python break and continue

• Python Pass

Python Functions

• Python Function
• Function Argument
• Python Recursion
• Anonymous Function
• Global, Local and Nonlocal
• Python Global Keyword
• Python Modules
• Python Package

Python Datatypes

• Python Numbers
• Python List
• Python Tuple
• Python String
• Python Dictionary

Python Files

• Python File Operation
• Python Directory
• Python Exception
• Exception Handling
• User-defined Exception

Python Object & Class

• Classes & Objects
• Python Inheritance
• Multiple Inheritance
• Operator Overloading

Python Advanced Topics

• Python Iterator
• Python Generator
• Python Closure
• Python Decorators
• Python Property
• Python RegEx

Python Date and time

• Python datetime Module
• Python datetime.strftime()
• Python datetime.strptime()
• Current date & time
• Get current time
• Timestamp to datetime
• Python time Module
• Python time.sleep()

Python Tutorials

Python pass Statement

Python Assert Statement

• Python Exception Handling
• List of Keywords in Python
• Python if...else Statement

In computer programming, the if statement is a conditional statement. It is used to execute a block of code only when a specific condition is met. For example,

Suppose we need to assign different grades to students based on their scores.

• If a student scores above 90 , assign grade A
• If a student scores above 75 , assign grade B
• If a student scores above 65 , assign grade C

These conditional tasks can be achieved using the if statement.

• Python if Statement

An if statement executes a block of code only if the specified condition is met.

Here, if the condition of the if statement is:

• True - the body of the if statement executes.
• False - the body of the if statement is skipped from execution.

Let's look at an example.

Note: Be mindful of the indentation while writing the if statements. Indentation is the whitespace at the beginning of the code.

Here, the spaces before the print() statement denote that it's the body of the if statement.

• Example: Python if Statement

Sample Output 1

In the above example, we have created a variable named number . Notice the test condition ,

As the number is greater than 0 , the condition evaluates True . Hence, the body of the if statement executes.

Sample Output 2

Now, let's change the value of the number to a negative integer, say -5 .

Now, when we run the program, the output will be:

This is because the value of the number is less than 0 . Hence, the condition evaluates to False . And, the body of the if statement is skipped.

An if statement can have an optional else clause. The else statement executes if the condition in the if statement evaluates to False .

Here, if the condition inside the if statement evaluates to

• True - the body of if executes, and the body of else is skipped.
• False - the body of else executes, and the body of if is skipped

• Example: Python if…else Statement

In the above example, we have created a variable named number .

Since the value of the number is 10 , the condition evaluates to True . Hence, code inside the body of if is executed.

If we change the value of the variable to a negative integer, let's say -5 , our output will be:

Here, the test condition evaluates to False . Hence code inside the body of else is executed.

• Python if…elif…else Statement

The if...else statement is used to execute a block of code among two alternatives.

However, if we need to make a choice between more than two alternatives, we use the if...elif...else statement.

• if condition1 - This checks if condition1 is True . If it is, the program executes code block 1 .
• elif condition2 - If condition1 is not True , the program checks condition2 . If condition2 is True , it executes code block 2 .
• else - If neither condition1 nor condition2 is True , the program defaults to executing code block 3 .

• Example: Python if…elif…else Statement

Since the value of the number is 0 , both the test conditions evaluate to False .

Hence, the statement inside the body of else is executed.

• Python Nested if Statements

It is possible to include an if statement inside another if statement. For example,

Here's how this program works.

More on Python if…else Statement

In certain situations, the if statement can be simplified into a single line. For example,

This code can be compactly written as

This one-liner approach retains the same functionality but in a more concise format.

Python doesn't have a ternary operator. However, we can use if...else to work like a ternary operator in other languages. For example,

can be written as

We can use logical operators such as and and or within an if statement.

Here, we used the logical operator and to add two conditions in the if statement.

We also used >= (comparison operator) to compare two values.

Logical and comparison operators are often used with if...else statements. Visit Python Operators to learn more.

Table of Contents

• Introduction

Video: Python if...else Statement

Sorry about that.

Related Tutorials

Python Tutorial

• Module 2: The Essentials of Python »
• Conditional Statements
• View page source

Conditional Statements 

There are reading-comprehension exercises included throughout the text. These are meant to help you put your reading to practice. Solutions for the exercises are included at the bottom of this page.

In this section, we will be introduced to the if , else , and elif statements. These allow you to specify that blocks of code are to be executed only if specified conditions are found to be true, or perhaps alternative code if the condition is found to be false. For example, the following code will square x if it is a negative number, and will cube x if it is a positive number:

Please refer to the “Basic Python Object Types” subsection to recall the basics of the “boolean” type, which represents True and False values. We will extend that discussion by introducing comparison operations and membership-checking, and then expanding on the utility of the built-in bool type.

Comparison Operations 

Comparison statements will evaluate explicitly to either of the boolean-objects: True or False . There are eight comparison operations in Python:

The first six of these operators are familiar from mathematics:

Note that = and == have very different meanings. The former is the assignment operator, and the latter is the equality operator:

Python allows you to chain comparison operators to create “compound” comparisons:

Whereas == checks to see if two objects have the same value, the is operator checks to see if two objects are actually the same object. For example, creating two lists with the same contents produces two distinct lists, that have the same “value”:

Thus the is operator is most commonly used to check if a variable references the None object, or either of the boolean objects:

Use is not to check if two objects are distinct:

bool and Truth Values of Non-Boolean Objects 

Recall that the two boolean objects True and False formally belong to the int type in addition to bool , and are associated with the values 1 and 0 , respectively:

Likewise Python ascribes boolean values to non-boolean objects. For example,the number 0 is associated with False and non-zero numbers are associated with True . The boolean values of built-in objects can be evaluated with the built-in Python command bool :

and non-zero Python integers are associated with True :

The following built-in Python objects evaluate to False via bool :

Zero of any numeric type: 0 , 0.0 , 0j

Any empty sequence, such as an empty string or list: '' , tuple() , [] , numpy.array([])

Empty dictionaries and sets

Thus non-zero numbers and non-empty sequences/collections evaluate to True via bool .

The bool function allows you to evaluate the boolean values ascribed to various non-boolean objects. For instance, bool([]) returns False wherease bool([1, 2]) returns True .

if , else , and elif 

We now introduce the simple, but powerful if , else , and elif conditional statements. This will allow us to create simple branches in our code. For instance, suppose you are writing code for a video game, and you want to update a character’s status based on her/his number of health-points (an integer). The following code is representative of this:

Each if , elif , and else statement must end in a colon character, and the body of each of these statements is delimited by whitespace .

The following pseudo-code demonstrates the general template for conditional statements:

In practice this can look like:

In its simplest form, a conditional statement requires only an if clause. else and elif clauses can only follow an if clause.

Similarly, conditional statements can have an if and an else without an elif :

Conditional statements can also have an if and an elif without an else :

Note that only one code block within a single if-elif-else statement can be executed: either the “if-block” is executed, or an “elif-block” is executed, or the “else-block” is executed. Consecutive if-statements, however, are completely independent of one another, and thus their code blocks can be executed in sequence, if their respective conditional statements resolve to True .

Reading Comprehension: Conditional statements

Assume my_list is a list. Given the following code:

What will happen if my_list is [] ? Will IndexError be raised? What will first_item be?

Assume variable my_file is a string storing a filename, where a period denotes the end of the filename and the beginning of the file-type. Write code that extracts only the filename.

my_file will have at most one period in it. Accommodate cases where my_file does not include a file-type.

"code.py" \(\rightarrow\) "code"

"doc2.pdf" \(\rightarrow\) "doc2"

"hello_world" \(\rightarrow\) "hello_world"

Inline if-else statements 

Python supports a syntax for writing a restricted version of if-else statements in a single line. The following code:

can be written in a single line as:

This is suggestive of the general underlying syntax for inline if-else statements:

The inline if-else statement :

The expression A if <condition> else B returns A if bool(<condition>) evaluates to True , otherwise this expression will return B .

This syntax is highly restricted compared to the full “if-elif-else” expressions - no “elif” statement is permitted by this inline syntax, nor are multi-line code blocks within the if/else clauses.

Inline if-else statements can be used anywhere, not just on the right side of an assignment statement, and can be quite convenient:

We will see this syntax shine when we learn about comprehension statements. That being said, this syntax should be used judiciously. For example, inline if-else statements ought not be used in arithmetic expressions, for therein lies madness:

Short-Circuiting Logical Expressions 

Armed with our newfound understanding of conditional statements, we briefly return to our discussion of Python’s logic expressions to discuss “short-circuiting”. In Python, a logical expression is evaluated from left to right and will return its boolean value as soon as it is unambiguously determined, leaving any remaining portions of the expression unevaluated . That is, the expression may be short-circuited .

For example, consider the fact that an and operation will only return True if both of its arguments evaluate to True . Thus the expression False and <anything> is guaranteed to return False ; furthermore, when executed, this expression will return False without having evaluated bool(<anything>) .

To demonstrate this behavior, consider the following example:

According to our discussion, the pattern False and short-circuits this expression without it ever evaluating bool(1/0) . Reversing the ordering of the arguments makes this clear.

In practice, short-circuiting can be leveraged in order to condense one’s code. Suppose a section of our code is processing a variable x , which may be either a number or a string . Suppose further that we want to process x in a special way if it is an all-uppercased string. The code

is problematic because isupper can only be called once we are sure that x is a string; this code will raise an error if x is a number. We could instead write

but the more elegant and concise way of handling the nestled checking is to leverage our ability to short-circuit logic expressions.

See, that if x is not a string, that isinstance(x, str) will return False ; thus isinstance(x, str) and x.isupper() will short-circuit and return False without ever evaluating bool(x.isupper()) . This is the preferable way to handle this sort of checking. This code is more concise and readable than the equivalent nested if-statements.

Reading Comprehension: short-circuited expressions

Consider the preceding example of short-circuiting, where we want to catch the case where x is an uppercased string. What is the “bug” in the following code? Why does this fail to utilize short-circuiting correctly?

Links to Official Documentation 

Truth testing

Boolean operations

Comparisons

‘if’ statements

Reading Comprehension Exercise Solutions: 

Conditional statements

If my_list is [] , then bool(my_list) will return False , and the code block will be skipped. Thus first_item will be None .

First, check to see if . is even contained in my_file . If it is, find its index-position, and slice the string up to that index. Otherwise, my_file is already the file name.

Short-circuited expressions

fails to account for the fact that expressions are always evaluated from left to right. That is, bool(x.isupper()) will always be evaluated first in this instance and will raise an error if x is not a string. Thus the following isinstance(x, str) statement is useless.

Python One Line Conditional Assignment

Problem : How to perform one-line if conditional assignments in Python?

Example : Say, you start with the following code.

You want to set the value of x to 42 if boo is True , and do nothing otherwise.

Let’s dive into the different ways to accomplish this in Python. We start with an overview:

Exercise : Run the code. Are all outputs the same?

Next, you’ll dive into each of those methods and boost your one-liner superpower !

Method 1: Ternary Operator

The most basic ternary operator x if c else y returns expression x if the Boolean expression c evaluates to True . Otherwise, if the expression c evaluates to False , the ternary operator returns the alternative expression y .

Let’s go back to our example problem! You want to set the value of x to 42 if boo is True , and do nothing otherwise. Here’s how to do this in a single line:

While using the ternary operator works, you may wonder whether it’s possible to avoid the ...else x part for clarity of the code? In the next method, you’ll learn how!

If you need to improve your understanding of the ternary operator, watch the following video:

You can also read the related article:

• Python One Line Ternary

Method 2: Single-Line If Statement

Like in the previous method, you want to set the value of x to 42 if boo is True , and do nothing otherwise. But you don’t want to have a redundant else branch. How to do this in Python?

The solution to skip the else part of the ternary operator is surprisingly simple— use a standard if statement without else branch and write it into a single line of code :

To learn more about what you can pack into a single line, watch my tutorial video “If-Then-Else in One Line Python” :

Method 3: Ternary Tuple Syntax Hack

A shorthand form of the ternary operator is the following tuple syntax .

Syntax : You can use the tuple syntax (x, y)[c] consisting of a tuple (x, y) and a condition c enclosed in a square bracket. Here’s a more intuitive way to represent this tuple syntax.

In fact, the order of the <OnFalse> and <OnTrue> operands is just flipped when compared to the basic ternary operator. First, you have the branch that’s returned if the condition does NOT hold. Second, you run the branch that’s returned if the condition holds.

Clever! The condition boo holds so the return value passed into the x variable is the <OnTrue> branch 42 .

Don’t worry if this confuses you—you’re not alone. You can clarify the tuple syntax once and for all by studying my detailed blog article.

Related Article : Python Ternary — Tuple Syntax Hack

Python One-Liners Book: Master the Single Line First!

Python programmers will improve their computer science skills with these useful one-liners.

Python One-Liners will teach you how to read and write “one-liners”: concise statements of useful functionality packed into a single line of code. You’ll learn how to systematically unpack and understand any line of Python code, and write eloquent, powerfully compressed Python like an expert.

The book’s five chapters cover (1) tips and tricks, (2) regular expressions, (3) machine learning, (4) core data science topics, and (5) useful algorithms.

Detailed explanations of one-liners introduce key computer science concepts and boost your coding and analytical skills . You’ll learn about advanced Python features such as list comprehension , slicing , lambda functions , regular expressions , map and reduce functions, and slice assignments .

You’ll also learn how to:

• Leverage data structures to solve real-world problems , like using Boolean indexing to find cities with above-average pollution
• Use NumPy basics such as array , shape , axis , type , broadcasting , advanced indexing , slicing , sorting , searching , aggregating , and statistics
• Calculate basic statistics of multidimensional data arrays and the K-Means algorithms for unsupervised learning
• Create more advanced regular expressions using grouping and named groups , negative lookaheads , escaped characters , whitespaces, character sets (and negative characters sets ), and greedy/nongreedy operators
• Understand a wide range of computer science topics , including anagrams , palindromes , supersets , permutations , factorials , prime numbers , Fibonacci numbers, obfuscation , searching , and algorithmic sorting

By the end of the book, you’ll know how to write Python at its most refined , and create concise, beautiful pieces of “Python art” in merely a single line.

Get your Python One-Liners on Amazon!!

While working as a researcher in distributed systems, Dr. Christian Mayer found his love for teaching computer science students.

To help students reach higher levels of Python success, he founded the programming education website Finxter.com that has taught exponential skills to millions of coders worldwide. He’s the author of the best-selling programming books Python One-Liners (NoStarch 2020), The Art of Clean Code (NoStarch 2022), and The Book of Dash (NoStarch 2022). Chris also coauthored the Coffee Break Python series of self-published books. He’s a computer science enthusiast, freelancer , and owner of one of the top 10 largest Python blogs worldwide.

His passions are writing, reading, and coding. But his greatest passion is to serve aspiring coders through Finxter and help them to boost their skills. You can join his free email academy here.

How to use python if else in one line with examples

How do I write a simple python if else in one line? What are ternary operator in Python? Can we use one liner for complex if and else statements?

In this tutorial I will share different examples to help you understand and learn about usage of ternary operator in one liner if and else condition with Python. Conditional expressions (sometimes called a “ ternary operator ”) have the lowest priority of all Python operations. Programmers coming to Python from C, C++, or Perl sometimes miss the so-called ternary operator ?:. It’s most often used for avoiding a few lines of code and a temporary variable for simple decisions.

I will not go into details of generic ternary operator as this is used across Python for loops and control flow statements. Here we will concentrate on learning python if else in one line using ternary operator

Python if else in one line

The general syntax of single if and else statement in Python is:

Now if we wish to write this in one line using ternary operator, the syntax would be:

In this syntax, first of all the else condition is evaluated.

• If condition returns True then value_when_true is returned
• If condition returns False then value_when_false is returned

Similarly if you had a variable assigned in the general if else block based on the condition

The same can be written in single line:

Here as well, first of all the condition is evaluated.

• if condition returns True then true-expr is assigned to value object
• if condition returns False then false-expr is assigned to value object

For simple cases like this, I find it very nice to be able to express that logic in one line instead of four. Remember, as a coder, you spend much more time reading code than writing it, so Python's conciseness is invaluable.

Some important points to remember:

• You can use a ternary expression in Python, but only for expressions , not for statements
• You cannot use Python if..elif..else block in one line.
• The name " ternary " means there are just 3 parts to the operator: condition , then , and else .
• Although there are hacks to modify if..elif..else block into if..else block and then use it in single line but that can be complex depending upon conditions and should be avoided
• With if-else blocks , only one of the expressions will be executed.
• While it may be tempting to always use ternary expressions to condense your code, realise that you may sacrifice readability if the condition as well as the true and false expressions are very complex.

Python Script Example

This is a simple script where we use comparison operator in our if condition

• First collect user input in the form of integer and store this value into b
• If b is greater than or equal to 0 then return " positive " which will be True condition
• If b returns False i.e. above condition was not success then return " negative "
• The final returned value i.e. either " positive " or " negative " is stored in object a
• Lastly print the value of value a

The multi-line form of this code would be:

Python if..elif..else in one line

Now as I told this earlier, it is not possible to use if..elif..else block in one line using ternary expressions. Although we can hack our way into this but make sure the maximum allowed length of a line in Python is 79 as per PEP-8 Guidelines

We have this if..elif..else block where we return expression based on the condition check:

We can write this if..elif..else block in one-line using this syntax:

In this syntax,

• First of all condition2 is evaluated, if return True then expr2 is returned
• If condition2 returns False then condition1 is evaluated, if return True then expr1 is returned
• If condition1 also returns False then else is executed and expr is returned

As you see, it was easier if we read this in multi-line if..elif..else block while the same becomes hard to understand for beginners.

We can add multiple if else block in this syntax, but we must also adhere to PEP-8 guidelines

Python Script Example-1

In this sample script we collect an integer value from end user and store it in " b ". The order of execution would be:

• If the value of b is less than 0 then " neg " is returned
• If the value of b is greater than 0 then " pos " is returned.
• If both the condition return False , then " zero " is returned

The multi-line form of the code would be:

Output(when if condition is True )

Output(when if condition is False and elif condition is True )

Output(when both if and elif condition are False )

Python script Example-2

We will add some more else blocks in this sample script, the order of the check would be in below sequence :

• Collect user input for value b which will be converted to integer type
• If value of b is equal to 100 then return " equal to 100 ", If this returns False then next if else condition would be executed
• If value of b is equal to 50 then return " equal to 50 ", If this returns False then next if else condition would be executed
• If value of b is equal to 40 then return " equal to 40 ", If this returns False then next if else condition would be executed
• If value of b is greater than 100 then return " greater than 100 ", If this returns False then next go to else block
• Lastly if all the condition return False then return " less than hundred "

The multi-line form of this example would be:

Python nested if..else in one line

We can also use ternary expression to define nested if..else block on one line with Python.

If you have a multi-line code using nested if else block , something like this:

The one line syntax to use this nested if else block in Python would be:

Here, we have added nested if..elif..else inside the else block using ternary expression. The sequence of the check in the following order

• If condition1 returns True then expr1 is returned, if it returns False then next condition is checked
• If condition-m returns True then expr-m is returned, if it returns False then else block with nested if..elif..else is checked
• If condition3 returns True then expr3 is returned, if it returns False then next condition inside the nested block is returned
• If condition-n returns True then expr-n is returned, if it returns False then expr5 is returned from the else condition

In this example I am using nested if else inside the else block of our one liner. The order of execution will be in the provided sequence:

• First of all collect integer value of b from the end user
• If the value of b is equal to 100 then the if condition returns True and " equal to 100 " is returned
• If the value of b is equal to 50 then the elif condition returns True and " equal to 50 " is returned
• If both if and elif condition returns False then the else block is executed where we have nested if and else condition
• Inside the else block , if b is greater than 100 then it returns " greater than 100 " and if it returns False then " less than 100 " is returned

In this tutorial we learned about usage of ternary operator in if else statement to be able to use it in one line. Although Python does not allow if..elif..else statement in one line but we can still break it into if else and then use it in single line form. Similarly we can also use nested if with ternary operator in single line. I shared multiple examples to help you understand the concept of ternary operator with if and else statement of Python programming language

Lastly I hope this tutorial guide on python if else one line was helpful. So, let me know your suggestions and feedback using the comment section.

Deepak Prasad

He is the founder of GoLinuxCloud and brings over a decade of expertise in Linux, Python, Go, Laravel, DevOps, Kubernetes, Git, Shell scripting, OpenShift, AWS, Networking, and Security. With extensive experience, he excels in various domains, from development to DevOps, Networking, and Security, ensuring robust and efficient solutions for diverse projects. You can reach out to him on his LinkedIn profile or join on Facebook page.

Can't find what you're searching for? Let us assist you.

Enter your query below, and we'll provide instant results tailored to your needs.

If my articles on GoLinuxCloud has helped you, kindly consider buying me a coffee as a token of appreciation.

For any other feedbacks or questions you can send mail to [email protected]

Thank You for your support!!

Leave a Comment Cancel reply

Save my name and email in this browser for the next time I comment.

Notify me via e-mail if anyone answers my comment.

How to Write the Python if Statement in one Line

• online practice

Have you ever heard of writing a Python if statement in a single line? Here, we explore multiple ways to do exactly that, including using conditional expressions in Python.

The if statement is one of the most fundamental statements in Python. In this article, we learn how to write the Python if in one line.

The if is a key piece in writing Python code. It allows developers to control the flow and logic of their code based on information received at runtime. However, many Python developers do not know they may reduce the length and complexity of their if statements by writing them in a single line.

For this article, we assume you’re somewhat familiar with Python conditions and comparisons. If not, don’t worry! Our Python Basics Course will get you up to speed in no time. This course is included in the Python Basics Track , a full-fledged Python learning track designed for complete beginners.

We start with a recap on how Python if statements work. Then, we explore some examples of how to write if statements in a single line. Let’s get started!

How the if Statement Works in Python

Let’s start with the basics. An if statement in Python is used to determine whether a condition is True or False . This information can then be used to perform specific actions in the code, essentially controlling its logic during execution.

The structure of the basic if statement is as follows:

The <expression> is the code that evaluates to either True or False . If this code evaluates to True, then the code below (represented by <perform_action> ) executes.

Python uses whitespaces to indicate which lines are controlled by the if statement. The if statement controls all indented lines below it. Typically, the indentation is set to four spaces (read this post if you’re having trouble with the indentation ).

As a simple example, the code below prints a message if and only if the current weather is sunny:

The if statement in Python has two optional components: the elif statement, which executes only if the preceding if/elif statements are False ; and the else statement, which executes only if all of the preceding if/elif statements are False. While we may have as many elif statements as we want, we may only have a single else statement at the very end of the code block.

Here’s the basic structure:

Here’s how our previous example looks after adding elif and else statements. Change the value of the weather variable to see a different message printed:

How to Write a Python if in one Line

Writing an if statement in Python (along with the optional elif and else statements) uses a lot of whitespaces. Some people may find it confusing or tiresome to follow each statement and its corresponding indented lines.

To overcome this, there is a trick many Python developers often overlook: write an if statement in a single line !

Though not the standard, Python does allow us to write an if statement and its associated action in the same line. Here’s the basic structure:

As you can see, not much has changed. We simply need to “pull” the indented line <perform_action> up to the right of the colon character ( : ). It’s that simple!

Let’s check it with a real example. The code below works as it did previously despite the if statement being in a single line. Test it out and see for yourself:

Writing a Python if Statement With Multiple Actions in one Line

That’s all well and good, but what if my if statement has multiple actions under its control? When using the standard indentation, we separate different actions in multiple indented lines as the structure below shows:

Can we do this in a single line? The surprising answer is yes! We use semicolons to separate each action in the same line as if placed in different lines.

Here’s how the structure looks:

And an example of this functionality:

Have you noticed how each call to the print() function appears in its own line? This indicates we have successfully executed multiple actions from a single line. Nice!

By the way, interested in learning more about the print() function? We have an article on the ins and outs of the print() function .

Writing a Full Python if/elif/else Block Using Single Lines

You may have seen this coming, but we can even write elif and else statements each in a single line. To do so, we use the same syntax as writing an if statement in a single line.

Here’s the general structure:

Looks simple, right? Depending on the content of your expressions and actions, you may find this structure easier to read and understand compared to the indented blocks.

Here’s our previous example of a full if/elif/else block, rewritten as single lines:

Using Python Conditional Expressions to Write an if/else Block in one Line

There’s still a final trick to writing a Python if in one line. Conditional expressions in Python (also known as Python ternary operators) can run an if/else block in a single line.

A conditional expression is even more compact! Remember it took at least two lines to write a block containing both if and else statements in our last example.

In contrast, here’s how a conditional expression is structured:

The syntax is somewhat harder to follow at first, but the basic idea is that <expression> is a test. If the test evaluates to True , then <value_if_true> is the result. Otherwise, the expression results in <value_if_false> .

As you can see, conditional expressions always evaluate to a single value in the end. They are not complete replacements for an if/elif/else block. In fact, we cannot have elif statements in them at all. However, they’re most helpful when determining a single value depending on a single condition.

Take a look at the code below, which determines the value of is_baby depending on whether or not the age is below five:

This is the exact use case for a conditional expression! Here’s how we rewrite this if/else block in a single line:

Much simpler!

Go Even Further With Python!

We hope you now know many ways to write a Python if in one line. We’ve reached the end of the article, but don’t stop practicing now!

If you do not know where to go next, read this post on how to get beyond the basics in Python . If you’d rather get technical, we have a post on the best code editors and IDEs for Python . Remember to keep improving!

You may also like

How Do You Write a SELECT Statement in SQL?

What Is a Foreign Key in SQL?

Enumerate and Explain All the Basic Elements of an SQL Query

Conditional expression (ternary operator) in Python

Python has a conditional expression (sometimes called a "ternary operator"). You can write operations like if statements in one line with conditional expressions.

• 6. Expressions - Conditional expressions — Python 3.11.3 documentation

Basics of the conditional expression (ternary operator)

If ... elif ... else ... by conditional expressions, list comprehensions and conditional expressions, lambda expressions and conditional expressions.

See the following article for if statements in Python.

• Python if statements (if, elif, else)

In Python, the conditional expression is written as follows.

The condition is evaluated first. If condition is True , X is evaluated and its value is returned, and if condition is False , Y is evaluated and its value is returned.

If you want to switch the value based on a condition, simply use the desired values in the conditional expression.

If you want to switch between operations based on a condition, simply describe each corresponding expression in the conditional expression.

An expression that does not return a value (i.e., an expression that returns None ) is also acceptable in a conditional expression. Depending on the condition, either expression will be evaluated and executed.

The above example is equivalent to the following code written with an if statement.

You can also combine multiple conditions using logical operators such as and or or .

• Boolean operators in Python (and, or, not)

By combining conditional expressions, you can write an operation like if ... elif ... else ... in one line.

However, it is difficult to understand, so it may be better not to use it often.

The following two interpretations are possible, but the expression is processed as the first one.

In the sample code below, which includes three expressions, the first expression is interpreted like the second, rather than the third:

By using conditional expressions in list comprehensions, you can apply operations to the elements of the list based on the condition.

See the following article for details on list comprehensions.

• List comprehensions in Python

Conditional expressions are also useful when you want to apply an operation similar to an if statement within lambda expressions.

In the example above, the lambda expression is assigned to a variable for convenience, but this is not recommended by PEP8.

Refer to the following article for more details on lambda expressions.

• Lambda expressions in Python

Related Categories

Related articles.

• Shallow and deep copy in Python: copy(), deepcopy()
• Composite two images according to a mask image with Python, Pillow
• OpenCV, NumPy: Rotate and flip image
• pandas: Check if DataFrame/Series is empty
• Check pandas version: pd.show_versions
• Python if statement (if, elif, else)
• pandas: Find the quantile with quantile()
• Handle date and time with the datetime module in Python
• Get image size (width, height) with Python, OpenCV, Pillow (PIL)
• Convert between Unix time (Epoch time) and datetime in Python
• Convert BGR and RGB with Python, OpenCV (cvtColor)
• Matrix operations with NumPy in Python
• pandas: Replace values in DataFrame and Series with replace()
• Uppercase and lowercase strings in Python (conversion and checking)
• Calculate mean, median, mode, variance, standard deviation in Python

Python If Else Statement – Conditional Statements Explained

There are many cases where you don't want all of your code to be executed in your programs.

Instead, you might want certain code to run only when a specific condition is met, and a different set of code to run when the condition is not satisified.

That's where conditional statements come in handy.

Conditional statements allow you to control the logical flow of programs in a clean and compact way.

They are branches – like forks in the road – that modify how code is executed and handle decision making.

This tutorial goes over the basics of if , if..else , and elif statements in the Python programming language, using examples along the way.

Let's get started!

The syntax of a basic if statement

An if statement in Python essentially says:

"If this expression evaluates to True, then run once the code that follows the exprerssion. If it isn't True, then don't run the block of code that follows."

The general syntax for a basic if statement looks something like this:

An if statement consists of:

• The if keyword, which starts the if statement.
• Then comes a condition. A condition can evaluate to either True or False. Parentheses ( () ) surrounding the condition are optional, but they do help improve the readability of the code when more than one condition is present.
• A colon : that separates the condition from the executable statement that follows.
• A new line.
• A level of indentation of 4 spaces, which is a Python convention. The level of indentation is associated with the body of the statement that follows.
• Lastly comes the body of the statement. This is the code that will run only if the statement evaluated to True. We can have multiple lines in the body that can be executed, and in that case we need to be careful they all have the same level of indentation.

Let's take the following example:

In the example above, we created two variables, a and b , and assigned them the values 1 and 2 , respectively.

The phrase in the print statement does in fact get printed to the console because the condition b > a evaluated to True, so the code that followed it ran. If it wasn't True, nothing would have happend. No code would have run.

If we had instead done this:

No code would have been executed and nothing would have been printed to the console.

How do Python if..else statements work?

An if statement runs code only when a condition is met. Nothing happens otherwise.

What if we also want code to run when the condition is not met? That's where the else part comes in.

The syntax of an if..else statement looks like this:

An if..else statement in Python means:

"When the if expression evaluates to True, then execute the code that follows it. But if it evalates to False, then run the code that follows the else statement"

The else statement is written on a new line after the last line of indented code and it can't be written by itself. An else statement is part of an if statement.

The code following it also needs to be indented with 4 spaces to show it is part of the else clause.

The code following the else statement gets executed if and only if the if statement is False. If your if statement is True and therefore the code ran, then the code in the else block will never run.

Here, the line of code following the else statement, print("a is in fact bigger than b") , will never run. The if statement that came before it is True so only that code runs instead.

The else block runs when:

Be aware that you can't write any other code between if and else . You'll get a SyntaxError if you do so:

How does elif work in Python?

What if we want to have more than just two options?

Instead of saying: "If the first condition is true do this, otherwise do that instead", now we say "If this isn't True, try this instead, and if all conditions fail to be True, resort to doing this".

elif stands for else, if.

The basic syntax looks like this:

We can use more than one elif statement. This gives us more conditions and more options.

For example:

In this example, the if statement tests a specific condition, the elif blocks are two alternatives, and the else block is the last solution when all the previous conditions have not been met.

Be aware of the order in which you write your elif statements.

In the previous example, if you had written:

The line x is less than 20! would have been executed because it came first.

The elif statement makes code easier to write. You can use it instead of keeping track of if..else statements as programs get more complex and grow in size.

If all the elif statements are not considered and are False, then and only then as the last resort will the code following the else statement run.

For example, here's a case when the else statement would run:

And that's it!

Those are the basic principles of if , if..else and elif in Python to get you started with conditional statements.

From here the statements can get more advanced and complex.

Conditional statements can be nested inside of other conditional statements, depending on the problem you're trying to solve and the logic behind the solution.

Thanks for reading and happy coding!

Read more posts .

If this article was helpful, share it .

Learn to code for free. freeCodeCamp's open source curriculum has helped more than 40,000 people get jobs as developers. Get started

Python If-Else Statement in One Line - Ternary Operator Explained

Single-line conditionals in python here’s when to and when not to use them.

Python isn’t the fastest programming language out there, but boy is it readable and efficient to write. Everyone knows what conditional statements are, but did you know you can write if statements in one line of Python code? As it turns out you can, and you’ll learn all about it today.

After reading, you’ll know everything about Python’s If Else statements in one line. You’ll understand when to use them, and when it’s best to avoid them and stick to conventional conditional statements.

Don’t feel like reading? Watch my video instead:

Want to get hired as a data scientist? Running a data science blog might help:

Can Blogging About Data Science Really Get You Hired as a Data Scientist?

What’s Wrong With the Normal If Statement?

Absolutely nothing. Splitting conditional statements into multiple lines of code has been a convention for ages. Most programming languages require the usage of curly brackets, and hence the single line if statements are not an option. Other languages allow writing only simple conditionals in a single line.

And then there’s Python. Before diving into If Else statements in one line, let’s first make a short recap on regular conditionals.

For example, you can check if a condition is true with the following syntax:

The variable age is less than 18 in this case, so Go home. is printed to the console. You can spice things up by adding an else condition that gets evaluated if the first condition is False :

This time age is greater than 18, so Welcome! gets printed to the console. Finally, you can add one or multiple elif conditions. These are used to capture the in-between cases. For example, you can print something entirely different if age is between 16 (included) and 18 (excluded):

The variable age is 17, which means the condition under elif is True , hence Not sure... is printed to the console.

Pretty basic stuff, so we naturally don’t want to spend so many lines of code writing it. As it turns out, you can use the ternary operator in Python to evaluate conditions in a single line.

Ternary Operator in Python

A ternary operator exists in some programming languages, and it allows you to shorten a simple If-Else block. It takes in 3 or more operands:

• Value if true - A value that’s returned if the condition evaluates to True.
• Condition - A boolean condition that has to be satisfied to return value if true.
• Value if false - A value that’s returned if the condition evaluates to False. In code, it would look like this:

You can even write else-if logic in Python’s ternary operator. In that case, the syntax changes slightly:

I have to admit - it looks a bit abstract when written like this. You’ll see plenty of practical examples starting from the next section.

One-Line If Statement (Without Else)

A single-line if statement just means you’re deleting the new line and indentation. You’re still writing the same code, with the only twist being that it takes one line instead of two.

Note: One-line if statement is only possible if there’s a single line of code following the condition. In any other case, wrap the code that will be executed inside a function.

Here’s how to transform our two-line if statement to a single-line conditional:

As before, age is less than 18 so Go home. gets printed.

What if you want to print three lines instead of one? As said before, the best practice is to wrap the code inside a function:

One-line if statements in Python are pretty boring. The real time and space saving benefit happens when you add an else condition.

You’ll benefit the most from one-line if statements if you add one or multiple else conditions.

One-Line If-Else Statement

Now we can fully leverage the power of Python’s ternary operator. The code snippet below stores Go home. to a new variable outcome if the age is less than 18 or Welcome! otherwise:

As you would guess, Welcome! is printed to the console as age is set to 19. If you want to print multiple lines or handle more complex logic, wrap everything you want to be executed into a function - just as before.

You now have a clear picture of how the ternary operator works on a simple one-line if-else statement. We can add complexity by adding more conditions to the operator.

One-Line If-Elif-Else Statement

Always be careful when writing multiple conditions in a single line of code. The logic will still work if the line is 500 characters long, but it’s near impossible to read and maintain it.

You should be fine with two conditions in one line, as the code is still easy to read. The following example prints Go home. if age is below 16, Not Sure... if age is between 16 (included) and 18 (excluded), and Welcome otherwise:

You’ll see Not sure... printed to the console, since age is set to 17. What previously took us six lines of code now only takes one. Neat improvement, and the code is still easy to read and maintain.

What else can you do with one-line if statements? Well, a lot. We’ll explore single-line conditionals for list operations next.

Example: One-Line Conditionals for List Operations

Applying some logic to a list involves applying the logic to every list item, and hence iterating over the entire list. Before even thinking about a real-world example, let’s see how you can write a conditional statement for every list item in a single line of code.

How to Write IF and FOR in One Line

You’ll need to make two changes to the ternary operator:

Surround the entire line of code with brackets [] Append the list iteration code (for element in array) after the final else Here’s how the generic syntax looks like:

It’s not that hard, but let’s drive the point home with an example. The following code snippet prints + if the current number of a range is greater than 5 and - otherwise. The numbers range from 1 to 10 (included):

Image 1 - If and For in a single line in Python (image by author)

Let’s now go over an additional real-world example.

Example: Did Student Pass the Exam?

To start, we’ll declare a list of students. Each student is a Python dictionary object with two keys: name and test score:

We want to print that the student has passed the exam if the score is 50 points or above. If the score was below 50 points, we want to print that the student has failed the exam.

In traditional Python syntax, we would manually iterate over each student in the list and check if the score is greater than 50:

Image 2 - List iteration with traditional Python syntax (image by author)

The code works, but we need 5 lines to make a simple check and store the results. You can use your newly-acquired knowledge to reduce the amount of code to a single line:

Image 3 - One-line conditional and a loop with Python (image by author)

The results are identical, but we have a much shorter and neater code. It’s just on the boundary of being unreadable, which is often a tradeoff with ternary operators and single-line loops. You often can’t have both readable code and short Python scripts.

Just because you can write a conditional in one line, it doesn’t mean you should. Readability is a priority. Let’s see in which cases you’re better off with traditional if statements.

Be Careful With One-Line Conditionals

Just because code takes less vertical space doesn’t mean it’s easier to read. Now you’ll see the perfect example of that claim.

The below snippet checks a condition for every possible grade (1-5) with a final else condition capturing invalid input. The conditions take 12 lines of code to write, but the entire snippet is extremely readable:

As expected, you’ll see Grade = 1 printed to the console, but that’s not what we’re interested in. We want to translate the above snippet into a one-line if-else statement with the ternary operator.

It’s possible - but the end result is messy and unreadable:

This is an example of an extreme case where you have multiple conditions you have to evaluate. It’s better to stick with the traditional if statements, even though they take more vertical space.

Take home point: A ternary operator with more than two conditions is just a nightmare to write and debug.

And there you have it - everything you need to know about one-line if-else statements in Python. You’ve learned all there is about the ternary operator, and how to write conditionals starting with a single if to five conditions in between.

Remember to keep your code simple. The code that’s easier to read and maintain is a better-written code at the end of the day. Just because you can cram everything into a single line, doesn’t mean you should. You’ll regret it as soon as you need to make some changes.

An even cleaner way to write long conditionals is by using structural pattern matching - a new feature introduced in Python 3.10. It brings the beloved switch statement to Python for extra readability and speed of development.

What do you guys think of one-line if-else statements in Python? Do you use them regularly or have you switched to structural pattern matching? Let me know in the comment section below.

• Learn Python
• Python Lists
• Python Dictionaries
• Python Strings
• Python Functions
• Learn Pandas & NumPy
• Pandas Tutorials
• Numpy Tutorials
• Learn Data Visualization
• Python Seaborn
• Python Matplotlib

Inline If in Python: The Ternary Operator in Python

• September 16, 2021 December 20, 2022

In this tutorial, you’ll learn how to create inline if statements in Python. This is often known as the Python ternary operator, which allows you to execute conditional if statements in a single line, allowing statements to take up less space and often be written in my easy-to-understand syntax! Let’s take a look at what you’ll learn.

The Quick Answer: Use the Python Ternary Operator

Table of Contents

What is the Python Ternary Operator?

A ternary operator is an inline statement that evaluates a condition and returns one of two outputs. It’s an operator that’s often used in many programming languages, including Python, as well as math. The Python ternary operator has been around since Python 2.5, despite being delayed multiple times.

The syntax of the Python ternary operator is a little different than that of other languages. Let’s take a look at what it looks like:

Now let’s take a look at how you can actually write an inline if statement in Python.

How Do you Write an Inline If Statement in Python?

Before we dive into writing an inline if statement in Python, let’s take a look at how if statements actually work in Python. With an if statement you must include an if , but you can also choose to include an else statement, as well as one more of else-ifs, which in Python are written as elif .

The traditional Python if statement looks like this:

This can be a little cumbersome to write, especially if you conditions are very simple. Because of this, inline if statements in Python can be really helpful to help you write your code faster.

Let’s take a look at how we can accomplish this in Python:

This is significantly easier to write. Let’s break this down a little bit:

• We assign a value to x , which will be evaluated
• We declare a variable, y , which we assign to the value of 10, if x is True. Otherwise, we assign it a value of 20.

We can see how this is written out in a much more plain language than a for-loop that may require multiple lines, thereby wasting space.

Tip! This is quite similar to how you’d written a list comprehension. If you want to learn more about Python List Comprehensions, check out my in-depth tutorial here . If you want to learn more about Python for-loops, check out my in-depth guide here .

Now that you know how to write a basic inline if statement in Python, let’s see how you can simplify it even further by omitting the else statement.

How To Write an Inline If Statement Without an Else Statement

Now that you know how to write an inline if statement in Python with an else clause, let’s take a look at how we can do this in Python.

Before we do this, let’s see how we can do this with a traditional if statement in Python

You can see that this still requires you to write two lines. But we know better – we can easily cut this down to a single line. Let’s get started!

We can see here that really what this accomplishes is remove the line break between the if line and the code it executes.

Now let’s take a look at how we can even include an elif clause in our inline if statements in Python!

Check out some other Python tutorials on datagy.io, including our complete guide to styling Pandas and our comprehensive overview of Pivot Tables in Pandas !

How to Write an Inline If Statement With an Elif Statement

Including an else-if, or elif , in your Python inline if statement is a little less intuitive. But it’s definitely doable! So let’s get started. Let’s imagine we want to write this if-statement:

Let’s see how we can easily turn this into an inline if statement in Python:

This is a bit different than what we’ve seen so far, so let’s break it down a bit:

• First, we evaluate is x == 1. If that’s true, the conditions end and y = 10.
• Otherwise, we create another condition in brackets
• First we check if x == 20, and if that’s true, then y = 20. Note that we did not repeated y= here.
• Finally, if neither of the other decisions are true, we assign 30 to y

This is definitely a bit more complex to read, so you may be better off creating a traditional if statement.

In this post, you learned how to create inline if statement in Python! You learned about the Python ternary operator and how it works. You also learned how to create inline if statements with else statements, without else statements, as well as with else if statements.

To learn more about Python ternary operators, check out the official documentation here .

Nik Piepenbreier

Nik is the author of datagy.io and has over a decade of experience working with data analytics, data science, and Python. He specializes in teaching developers how to use Python for data science using hands-on tutorials. View Author posts

Leave a Reply Cancel reply

Your email address will not be published. Required fields are marked *

Save my name, email, and website in this browser for the next time I comment.

Assignments  »  Conditional Structures  »  Set1

Conditional Structures

1. Write a program that prompts the user to input a number and display if the number is even or odd. Solution

2. Write a Python program that takes an age as input and determines whether a person is eligible to vote. If the age is 18 or above, print "You are eligible to vote." Otherwise, print "You are not eligible to vote yet.". Solution

3. Write a program that prompts the user to input two integers and outputs the largest. Solution

4. Write a program that prompts the user to enter a number and determines whether it is positive, negative, or zero. The program should print "Positive" if the number is greater than 0, "Negative" if the number is less than 0, and "Zero" if the number is 0. Solution

5. Write a program that prompts the user to enter their age and prints the corresponding age group. The program should use the following age groups:

6. Write a program that prompts the user to input a number from 1 to 7. The program should display the corresponding day for the given number. For example, if the user types 1, the output should be Sunday. If the user types 7, the output should be Saturday. If the number is not between 1 to 7 user should get error message as shown in sample output. Solution

7. Write a program that prompts the user to enter their weight (in kilograms) and height (in meters). The program should calculate the Body Mass Index (BMI) using the formula: BMI = weight / (height * height). The program should then classify the BMI into one of the following categories:

8. The marks obtained by a student in 3 different subjects are input by the user. Your program should calculate the average of subjects and display the grade. The student gets a grade as per the following rules:

Write a program that prompts the user to input the value of a (the coefficient of x 2 ), b (the coefficient of x), and c (the constant term) and outputs the roots of the quadratic equation. Solution

10. Write a program that prompts the user to enter three numbers and sorts them in ascending order. The program should print the sorted numbers. Solution

11. Write a program that prompts the user to input three integers and outputs the largest. Solution

12. Write a program that prompts the user to input a character and determine the character is vowel or consonant. Solution

13. Write a program that prompts the user to input a year and determine whether the year is a leap year or not. Leap Years are any year that can be evenly divided by 4. A year that is evenly divisible by 100 is a leap year only if it is also evenly divisible by 400. Example :

14. Write a program that prompts the user to input number of calls and calculate the monthly telephone bills as per the following rule: Minimum Rs. 200 for up to 100 calls. Plus Rs. 0.60 per call for next 50 calls. Plus Rs. 0.50 per call for next 50 calls. Plus Rs. 0.40 per call for any call beyond 200 calls. Solution

• Data Structures
• Linked List
• Binary Tree
• Binary Search Tree
• Segment Tree
• Disjoint Set Union
• Fenwick Tree
• Red-Black Tree
• Advanced Data Structures
• Shortest path with one curved edge in an undirected Graph
• Find the largest sum of a cycle in the maze
• Determining topology formed in a Graph
• Check whether the structure of given Graph is same as the game of Rock-Paper-Scissor
• Check if incoming edges in a vertex of directed graph is equal to vertex itself or not
• Hypercube Graph
• Construct a graph from given degrees of all vertices
• A Peterson Graph Problem
• 2-Satisfiability (2-SAT) Problem
• Number of sink nodes in a graph
• Largest subset of Graph vertices with edges of 2 or more colors
• Check whether given degrees of vertices represent a Graph or Tree
• Roots of a tree which give minimum height
• Push Relabel Algorithm | Set 2 (Implementation)
• Optimal read list for given number of days
• Channel Assignment Problem
• Graph Coloring Using Greedy Algorithm
• Introduction to Graph Coloring
• Find maximum number of edge disjoint paths between two vertices

Graph Coloring Algorithm in Python

Given an undirected graph represented by an adjacency matrix. The graph has  n   nodes, labeled from  1   to  n . The task is to assign colors to each node in such a way that no two adjacent nodes have the same color. The challenge is to solve this problem using the minimum number of colors.

Graph Coloring in Python using Greedy Algorithm :

The greedy graph coloring algorithm works by assigning colors to vertices one at a time, starting from the first vertex . It checks if any neighboring vertices share the same color before coloring a vertex. If a color can be assigned without clashing with neighbors, it’s considered a valid part of the solution. The algorithm continues until all vertices are colored. If a vertex can’t be validly colored, the algorithm backtracks and concludes that the graph can’t be colored with the available colors.

Step-by-step algorithm:

• Initialize an array result[] to store the assigned colors for each vertex. Initially, all vertices are assigned -1 indicating no color.
• Initialize an array available[] to keep track of available colors for each vertex.
• Assign the first color (let’s say 0 ) to the first vertex ( 0 ).
• Mark colors of adjacent vertices of u as unavailable in the available[] array.
• Find the first available color for u that is not used by its adjacent vertices.
• Assign the found color to vertex u .
• Print the assigned colors for each vertex.

Here is the implementation of the above idea:

Time Complexity : O(V * (V + E)), where V is the number of vertices and E is the number of edges. Auxiliary Space : O(V + E)

Graph Coloring in Python using Backtracking :

Assign colors one by one to different vertices, starting from vertex  0 . Before assigning a color, check if the adjacent vertices have the same color or not. If there is any color assignment that does not violate the conditions, mark the color assignment as part of the solution. If no assignment of color is possible then backtrack and return false .

• Create a recursive function that takes the graph, current index, number of vertices, and color array.
• If the current index is equal to the number of vertices. Print the color configuration in the color array.
• For every assigned color, check if the configuration is safe, (i.e. check if the adjacent vertices do not have the same color) and recursively call the function with the next index and number of vertices else return  false
• If any recursive function returns  true  then break the loop and return true
• If no recursive function returns  true  then return  false

Time Complexity : O(m V ), where V is the number of vertices and m is the number of available colors. Auxiliary Space : O(V)

Graph Coloring in Python using Branch & Bound Algorithm :

Graph coloring is a classic problem in computer science and graph theory, aiming to assign colors to vertices of a graph in such a way that no two adjacent vertices share the same color. One approach to solve this problem is the Branch and Bound algorithm . This algorithm systematically explores the search space of possible colorings while efficiently pruning branches of the search tree that are known to lead to suboptimal solutions .

• Create a graph object with vertices number of vertices and an empty adjacency list self.graph .
• Use add_edge() to add edges between vertices.
• Implement is_safe() to verify if assigning a color violates constraints.
• Implement graph_coloring_util() to recursively color vertices.
• Start at v = 0 and assign colors.
• Return True if successful, else backtrack .
• Implement graph_coloring() as the main function.
• Initialize color list.
• Call graph_coloring_util() to color the graph.
• Print colors if successful, else print message.

Please Login to comment...

Similar reads.

• Graph Coloring
• How to Use ChatGPT with Bing for Free?
• 7 Best Movavi Video Editor Alternatives in 2024
• How to Edit Comment on Instagram
• 10 Best AI Grammar Checkers and Rewording Tools
• 30 OOPs Interview Questions and Answers (2024)

Improve your Coding Skills with Practice

What kind of Experience do you want to share?