Skip to content
Related Articles

Related Articles

Improve Article

Itertools in Python3

  • Last Updated : 10 Jul, 2020

Itertools is a module in Python, it is used to iterate over data structures that can be stepped over using a for-loop. Such data structures are also known as iterables. This module works as a fast, memory-efficient tool that is used either by themselves or in combination to form iterator algebra.

Why to use ?

This module incorporates functions that utilize computational resources efficiently. Using this module also tends to enhance the readability and maintainability of the code.

grouper Recipe

The grouper() function can be found in the Recipes section of the itertools docs. The recipes are an excellent source of inspiration for ways to use itertools to your advantage.

Example




# Python code to demonstrate the 
# grouper Recipie
import itertools as it
  
# defining the grouper function
def grouper(inputs, n, fillvalue = None):
    iters = [iter(inputs)] * n
    return it.zip_longest(*iters, fillvalue = fillvalue)
  
alpha = ['g', 'e', 'e', 'k', 's', 'f', 'o',
         'r', 'g', 'e', 'e', 'k', 's']
print(list(grouper(alpha, 3)))

Output :



[(‘g’, ‘e’, ‘e’), (‘k’, ‘s’, ‘f’), (‘o’, ‘r’, ‘g’), (‘e’, ‘e’, ‘k’), (‘s’, None, None)]

Brute force scenario

Brute force is a straightforward method of solving a problem that relies on sheer computing power and trying every possibility rather than advanced techniques to improve efficiency. There are different Brute force itertools function such as:

  • combinations()
  • combinations_with_replacement()
  • permutations()

combinations()

The itertools.combinations() function takes two arguments—an iterable inputs and a positive integer n—and produces an iterator over tuples of all combinations of n elements in inputs.

Example




# Python code to demonstrate combinations
import itertools as it
  
print(list(it.combinations([1, 2], 2)))

Output :

[(1, 2)]

combinations_with_replacement()

combinations_with_replacement() works just like combinations(), accepting an iterable inputs and a positive integer n, and returns an iterator over n-tuples of elements from inputs. The difference is that combinations_with_replacement() allows elements to be repeated in the tuples it returns.
Example




# Python code to demonstrate combinations_with_replacement
import itertools as it
  
print(list(it.combinations_with_replacement([1, 2], 2)))

Output :

[(1, 1), (1, 2), (2, 2)] 

permutations()

A permutation is a collection or a combination of objects from a set where the order or the arrangement of the chosen objects does matter. permutations() accepts a single iterable and produces all possible permutations (rearrangements) of its elements.
Example




# Python code to demonstrate permutations
import itertools as it
  
print(list(it.permutations(['g', 'e', 'k'])))

Output :

[(‘g’, ‘e’, ‘k’), (‘g’, ‘k’, ‘e’), (‘e’, ‘g’, ‘k’), (‘e’, ‘k’, ‘g’), (‘k’, ‘g’, ‘e’), (‘k’, ‘e’, ‘g’)]

Flattening A List of Lists

Converting a list of lists (2D), into a list (1D) is called flattening. Flattening a list of lists merges all the sublists into one unified list.
Example




# Python code to demonstrate flattening a list of lists
import itertools as it
  
list_of_lists = [[1, 2], [3, 4]]
chain_object = it.chain.from_iterable(list_of_lists)
  
# Return chain object with nested lists separated
# Convert to list to flatten
flattened_list = list(chain_object)
  
print(flattened_list)

Output :

[1, 2, 3, 4] 

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course




My Personal Notes arrow_drop_up
Recommended Articles
Page :