Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ShyamaSankar/59b64532762c01310c56ef4b3c0226c4 to your computer and use it in GitHub Desktop.
Save ShyamaSankar/59b64532762c01310c56ef4b3c0226c4 to your computer and use it in GitHub Desktop.
List comprehension with an if-else condition.
# Original list of numbers.
numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
# For loop to create a list with squares of all odd numbers
# and doubles of all even numbers in the original list.
modified_numbers = []
for number in numbers:
if number % 2 == 1:
modified_numbers.append(number * number)
else:
modified_numbers.append(number * 2)
# Rewrite using list comprehension.
# Syntax:
# list_object = [expression_1_on_item if_condition_1_on_item else expression_2_on_item for_item_in_iterable]
modified_numbers = [number * number if number % 2 == 1 else number * 2 for number in numbers]
print(modified_numbers) # Output: [1, 4, 9, 8, 25, 12, 49, 16, 81, 20]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment