Python | Find top K frequent elements from a list of tuples

Given a list of tuples with word as first element and its frequency as second element, the task is to find top k frequent element.

Below are some ways to above achieve the above task.

Method #1: Using defaultdict

# Python code to find top 'k' frequent element

# Importing
import collections
from operator import itemgetter
from itertools import chain

# Input list initialization
Input =[[('Name', 151)], [('ACe', 400)],
        [('TURN', 210)], [('RED', 1113)],
        [('YELLOW', 1)]]

# K initialization
K = 3

# Using defaultdict to  find top 'k' frequent element
dict_ = collections.defaultdict(list)
new_list = list(chain.from_iterable(Input))

for elem in new_list:
    dict_[elem[0]].append(elem[1])

res = {k: sum(v) for k, v in dict_.items()}

# Using sorted
Output = sorted(res.items(), key = itemgetter(1),
                             reverse = True)[0:K]

# printing output
print("Initial List of tuple is", Input)
print("\nTop 'K' elements are", Output)

Output:

Initial List of tuple is [[(‘Name’, 151)], [(‘ACe’, 400)], [(‘TURN’, 210)], [(‘RED’, 1113)], [(‘YELLOW’, 1)]]

Top ‘K’ elements are [(‘RED’, 1113), (‘ACe’, 400), (‘TURN’, 210)]

Method #2: Using itertools and sorted

# Python code to find top 'k' frequent element
from operator import itemgetter
from itertools import chain

# Input list initialization
Input =[[('Name', 151)], [('ACe', 400)],
        [('TURN', 210)], [('RED', 1113)],
        [('YELLOW', 1)]]

# k initialization
K = 3

# Finding top 'k' frequent element
# without using collection
Output = sorted(list(chain.from_iterable(Input)),
        key = itemgetter(1), reverse = True)[0:K]

# Printing Output
print("Initial List of tuple is", Input)
print("\nTop 'K' elements are", Output)

Output:

Initial List of tuple is [[(‘Name’, 151)], [(‘ACe’, 400)], [(‘TURN’, 210)], [(‘RED’, 1113)], [(‘YELLOW’, 1)]]

Top ‘K’ elements are [(‘RED’, 1113), (‘ACe’, 400), (‘TURN’, 210)]