Python - Maximum of K element in other list

Last Updated : 02 May, 2023

Given two lists, extract maximum of elements with similar K in corresponding list.

Input : test_list1 = [4, 3, 6, 2, 8], test_list2 = [3, 6, 3, 4, 3], K = 3
Output : 8
Explanation : Elements corresponding to 3 are, 4, 6, and 8, Max. is 8.

Input : test_list1 = [10, 3, 6, 2, 8], test_list2 = [5, 6, 5, 4, 5], K = 5
Output : 10
Explanation : Elements corresponding to 5 are, 10, 6, and 8, Max. is 10.

Method #1 : Using loop + max()

In this, we extract all elements from list 1 which are equal to K in list 2, and then perform max() to get maximum of them.

Python3

# Python3 code to demonstrate working of 
# Maximum of K element in other list
# Using loop + max()

# initializing lists
test_list1 = [4, 3, 6, 2, 9]
test_list2 = [3, 6, 3, 4, 3]

# printing original lists
print("The original list 1 is : " + str(test_list1))
print("The original list 2 is : " + str(test_list2))

# initializing K 
K = 3

res = []
for idx in range(len(test_list1)):
    
    # checking for K in 2nd list
    if test_list2[idx] == K :
        res.append(test_list1[idx])

# getting Maximum element
res = max(res)

# printing result 
print("Extracted Maximum element : " + str(res))

Output

The original list 1 is : [4, 3, 6, 2, 9]
The original list 2 is : [3, 6, 3, 4, 3]
Extracted Maximum element : 9

Method #2 : list comprehension + max() + zip()

In this, we perform task of pairing elements using zip() and is one-liner solution provided using list comprehension.

Python3

# Python3 code to demonstrate working of 
# Maximum of K element in other list
# Using list comprehension + max() + zip()

# initializing lists
test_list1 = [4, 3, 6, 2, 9]
test_list2 = [3, 6, 3, 4, 3]

# printing original lists
print("The original list 1 is : " + str(test_list1))
print("The original list 2 is : " + str(test_list2))

# initializing K 
K = 3

# one liner to solve this problem
res = max([sub1 for sub1, sub2 in zip(test_list1, test_list2) if sub2 == K])

# printing result 
print("Extracted Maximum element : " + str(res))

Output

The original list 1 is : [4, 3, 6, 2, 9]
The original list 2 is : [3, 6, 3, 4, 3]
Extracted Maximum element : 9

Method #3 : Using heapq

Approach

Using heapq.nlargest() to find the max number

Algorithm

1. Create a list of tuples where each tuple contains elements from both test_list1 and test_list2 at corresponding indices.
2. Create a new list using list comprehension to filter out tuples that have the element K in test_list2.
3. Use heapq.nlargest() function to return the K largest elements from the filtered list.
4. Return the first element from the resulting list.

Python3

import heapq

def max_k_element_3(test_list1, test_list2, K):
    zipped_list = list(zip(test_list1, test_list2))
    filtered_list = [x[0] for x in zipped_list if x[1] == K]
    return heapq.nlargest(K, filtered_list)[0]
test_list1 = [4, 3, 6, 2, 9]
test_list2 = [3, 6, 3, 4, 3]
K=3
print(max_k_element_3(test_list1, test_list2, K))

Output

Time complexity: O(n log K) - due to using heapq.nlargest() function
Space complexity: O(n) - for creating a list of tuples

Method #4: Using numpy

Step-by-step approach:

Import the numpy library.
Convert both lists into numpy arrays using the np.array() method.
Use the np.where() method to find the indices of elements in test_list2 that are equal to K.
Use the resulting indices to slice the corresponding elements in test_list1 and store them in a new numpy array.
Use the np.amax() method to find the maximum value in the new array.
Convert the result back to a regular Python integer using the int() method.
Print the final result.

Python3

import numpy as np

# initializing lists
test_list1 = [4, 3, 6, 2, 9]
test_list2 = [3, 6, 3, 4, 3]

# printing original lists
print("The original list 1 is : " + str(test_list1))
print("The original list 2 is : " + str(test_list2))

# initializing K 
K = 3

# convert lists to numpy arrays
arr1 = np.array(test_list1)
arr2 = np.array(test_list2)

# find indices where arr2 == K
indices = np.where(arr2 == K)[0]

# slice arr1 using indices and store in new array
new_arr = arr1[indices]

# find maximum value in new array
max_val = np.amax(new_arr)

# convert result to integer
res = int(max_val)

# printing result 
print("Extracted Maximum element : " + str(res))

OUTPUT :
The original list 1 is : [4, 3, 6, 2, 9]
The original list 2 is : [3, 6, 3, 4, 3]
Extracted Maximum element : 9

Time complexity: O(n), where n is the length of the input lists.
Auxiliary space: O(n), for the numpy arrays and other variables created in the process.

Python Introduction

manjeet_04

Improve

Article Tags :

Practice Tags :

python

Python - Maximum of K element in other list

Method #3 : Using heapq

Approach

Algorithm

Similar Reads

Python Fundamentals

Python Data Structures

Advanced Python

Data Science with Python

Web Development with Python

Python Practice

Thank You!

What kind of Experience do you want to share?