Skip to main content

Python List Comprehension

Way to create a new list from iterable by transforming the items.

Syntax:


[ output_expression for item in iterable [ conditions ] ]

numbers = [1, 2, 3, 4, 5, 6]
squares = [number**2 for number in numbers]
print(squares)
#Output: [1, 4, 9, 16, 25, 36]

Using if condition


We can use if condition to filter out some elements.

numbers = [1, 2, 3, 4, 5, 6]
squares = [number**2 for number in numbers if number > 2]
print(squares)
#Output: [9, 16, 25, 36]

Using multiple if condition (ANDing)

We can use multiple if conditions to filter out elements. All if conditions will be ANDed

numbers = [1, 2, 3, 4, 5, 18, 30]
list_1 = [number for number in numbers
          if number % 2 == 0
          if number % 3 == 0]
print(list_1)
#Output: [18, 30]

Using ‘in’ statement (ORing)


'in' keyword can be used to create a condition statement.

numbers = [1, 2, 3, 4, 5, 18, 30]
list_2 = [number for number in numbers
          if number % 3 in (1, 2)]
print(list_2)
#Output: [1, 2, 4, 5]

Using if condition in output expression



If condition can be used to alter the resulting list.

numbers = [1, 2, 3, 4, 5]
list_3 = ["Even" if number % 2 == 0
          else "Odd" for number in numbers]
print(list_3)
#Output: ['Odd', 'Even', 'Odd', 'Even', 'Odd']

Using the if-else statement


Ladder of if-else statement can be used to alter the resulting list.

numbers = [25, 35, 50, 90]
list_4 = ["First" if number >= 60
          else "Second" if 45 <= number < 60
          else "Third" if 33 <= number < 45
          else "Failed"
          for number in numbers]
print(list_4)
# Output: ['Failed', 'Third', 'Second', 'First']

Nesting list comprehension


Nesting of list comprehension is just like the nested for loops.

list_6 = [[n for n in range(1, 4)] for number in range(3)]
print(list_6)
# Output: [[1, 2, 3], [1, 2, 3], [1, 2, 3]]

Dictionary comprehension


Dictionary comprehension works the same as lists. The difference is that we will work on (key, value) pairs.

dict_1 = {'a': 1, 'b': 3, 'c': 4}
dict_2 = {key: value**2 for key, value in dict_1.items()}
print(dict_2)
# Output: {'a': 2, 'b': 6, 'c': 8}

Set comprehension


Set can be created just like the list is comprehended.

numbers = [1, 2, 3, 4, 5]
set_1 = {number*2 for number in numbers}
print(set_1)
# Output: {2, 4, 6, 8, 10}

If you have some cool tricks regarding this topic, feel free to comment below. Thanks.
Labels:

Comments

Post a Comment

Popular posts from this blog

Insertion at specific position N in singly link list

Insertion of a new node at position N in singly link list requires traversing the list for N-1 th node so that links are updated to accommodate the new node in the list. The new node's next is set to point to Nth node of the list and then N-1 th node's next pointer is updated so that it reference to new node. The step by step algorithm to insert node at Nth position is as below: Algorithm to insert node at specific position N in singly link list: This algorithm will insert the new node PTR at the position N in the link list. The steps are as follows: Create new node PTR Set the INFO field of PTR If N is less than 1 Node can't be inserted Else If node is to be inserted at first i.e. [N=1] Make new node PTR points to first node i.e. [PTR->NEXT = START] Make START point to new node PTR i.e. [START = PTR] Else Traverse the list to get the (N-1)th node of list into TEMP Make PTR's next pointer point to Nth node in the list i.e. [PTR -> NEXT  =  TEMP-...
Post a Comment
Read more

Prefix to Infix Conversion

With a given Prefix Expression, we will see how to convert Prefix Expression into Infix Expression using stack.   Algorithm to convert Prefix Expression to Infix Expression: In this algorithm, we will use stack to store operands during the conversion. The step are as follows: Read the prefix string While the end of prefix string scanned from right to left symb = the current character If symb is an operator poped_sym1 = pop the stack poped_sym2 = pop the stack concat the string  STR = ( poped_sym1 )+ ( operator )+( poped_sym2 ) push the string STR into stack Else push the operand symb into stack End If End While infix_str = pop the stack   Function to convert Prefix Expression to Infix Expression: void prefix_to_infix(char prefix[], char infix[]){ char op[2]; //operator string char poped1[MAX]; char poped2[MAX]; char temp[MAX]; int i = strlen(prefix); op[1] = '\0'; while(--i != -1){ if(prefix[i] == ' '){ continue; } if(isoper...
Post a Comment
Read more

Link List

In computer science, a linked list is a linear collection of elements, called nodes, each node have info field and next field pointing to the next node by means of a pointer. It is a data structure consisting of a group of nodes in sequence and connected with links. In simplest form of this data structure, each node is consist of data and a reference (in other words, a link) to the next node in the sequence. This structure allows for efficient insertion or removal of elements from any position in the sequence during iteration. Types of Link List Singly Link List: Insert node at first position in singly link list Insert node at last position in singly link list Insert node at specific position in singly link list Delete node at first position in singly link list Delete node at last position in singly link list Delete node at specific position in singly link list Find number of nodes in singly link list Dublication of singly link list Concatenation of two singly link list So...
Post a Comment
Read more