Python - Conditional Prefix in List

Given a list of elements, attach different prefix according to condition.

Input : test_list = [45, 53, 76, 86, 3, 49], pref_1 = "LOSE-", pref_2 = "WIN-" 
Output : ['LOSE-45', 'WIN-53', 'WIN-76', 'WIN-86', 'LOSE-3', 'LOSE-49'] 
Explanation : All 50+ are prefixed as "WIN-" and others as "LOSE-".

Input : test_list = [78, 53, 76, 86, 83, 69], pref_1 = "LOSE-", pref_2 = "WIN-" 
Output : ['WIN-78', 'WIN-53', 'WIN-76', 'WIN-86', 'WIN-83', 'WIN-69'] 
Explanation : All are 50+ hence prefixed "WIN-". 

Method #1: Using loop

This brute way in which this task can be performed. In this, we perform the task of attaching the prefix by using conditionary operator and loop.

Step-by-step approach :

1. Initialize a list named test_list containing some integer values.
2. Print the original list using the print() function and string concatenation to display a message along with the list.
3. Initialize two prefixes, pref_1 and pref_2, as "LOW-" and "HIGH-" respectively.
4. Create an empty list named res.
5. Use a for loop to iterate over each element ele in the test_list.
6. For each element, we check if it is greater than or equal to 50 using an if statement.
   1. If the element is greater than or equal to 50, we append the prefix pref_2 followed by the element converted to a string using the str() function, to the res list using the append() function.
   2. If the element is less than 50, we append the prefix pref_1 followed by the element converted to a string using the str() function, to the res list using the append() function.
7. After iterating through all the elements in test_list, we print the res list containing the prefixed elements using the print() function and string concatenation to display a message along with the list.

# Python3 code to demonstrate working of 
# Conditional Prefix in List
# Using loop

# initializing list
test_list = [45, 53, 76, 86, 3, 49]

# printing original list
print("The original list : " + str(test_list))

# initializing pref 1
pref_1 = "LOW-"

# initializing pref 2
pref_2 = "HIGH-"

res = []
for ele in test_list:
    
    # appending prefix on greater than 50 check
    if ele >= 50:
        res.append(pref_2 + str(ele))
    else :
        res.append(pref_1 + str(ele))
        
# printing result 
print("The prefixed elements : " + str(res))

The original list : [45, 53, 76, 86, 3, 49]
The prefixed elements : ['LOW-45', 'HIGH-53', 'HIGH-76', 'HIGH-86', 'LOW-3', 'LOW-49']

Time complexity: O(n), where n is the length of the input list 'test_list'. This is because the program iterates through the entire list once and performs a constant amount of operations (appending the prefix to the element and appending the result to the 'res' list) for each element.
Auxiliary space: O(n), as it creates a new list 'res' of the same size as the input list 'test_list' to store the prefixed elements.

Method #2: Using list comprehension

This is one of the ways in which this task can be performed. In this, we perform the similar task as above as one liner using list comprehension.

Here are the steps involved:

1. First, a list of integers is initialized and assigned to the variable test_list.
2. The original list is printed using the print() function and a string concatenation.
3. Two string prefixes are initialized using the variables pref_1 and pref_2. pref_1 is set to "LOW-" and pref_2 is set to "HIGH-".
4. The list comprehension statement is used to create a new list named res. This statement loops through each element in the test_list and adds a conditional prefix based on the value of the element. If the element is greater than or equal to 50, the prefix is set to pref_2 and if it's less than 50, the prefix is set to pref_1. The resulting element is converted to a string using the str() function, and added to the new list res.
5. Finally, the resulting list res is printed using the print() function and a string concatenation.

# Python3 code to demonstrate working of 
# Conditional Prefix in List
# Using list comprehension

# initializing list
test_list = [45, 53, 76, 86, 3, 49]

# printing original list
print("The original list : " + str(test_list))

# initializing pref 1
pref_1 = "LOW-"

# initializing pref 2
pref_2 = "HIGH-"

# solution encapsulated as one-liner and conditional checks
res = [pref_2 + str(ele) if ele >= 50 else pref_1 + str(ele) for ele in test_list]
        
# printing result 
print("The prefixed elements : " + str(res))

The original list : [45, 53, 76, 86, 3, 49]
The prefixed elements : ['LOW-45', 'HIGH-53', 'HIGH-76', 'HIGH-86', 'LOW-3', 'LOW-49']

Time Complexity: O(n)
Auxiliary Space: O(n)

Method #3: Using map and lambda functions

 we are using the map function to apply the lambda function to each element of the test_list. The lambda function checks if the element is greater than or equal to 50 and adds the corresponding prefix to it

Steps:

1. Initialize the input list 'test_list' with some integers.
2. Print the original list using the 'print' function and concatenation.
3. Initialize the string variables 'pref_1' and 'pref_2' with the prefix strings 'LOW-' and 'HIGH-' respectively.
4. Use the 'map' function with a 'lambda' function to iterate over each element in the 'test_list' and perform the following operations:
   a. If the current element is greater than or equal to 50, then add the 'pref_2' string to the start of the string representation of the current element, otherwise add the 'pref_1' string to the start of the string representation of the current element.
   b. Append the resulting string to the output list 'res'.
5. Print the resulting list 'res' using the 'print' function and concatenation. 

# Python3 code to demonstrate working of
# Conditional Prefix in List
# Using map and lambda

test_list = [45, 53, 76, 86, 3, 49]

# printing original list
print("The original list : " + str(test_list))

pref_1 = "LOW-"

pref_2 = "HIGH-"

res = list(map(lambda x: pref_2 + str(x) if x >=
               50 else pref_1 + str(x), test_list))

# printing the result
print("The prefixed elements : " + str(res))

The original list : [45, 53, 76, 86, 3, 49]
The prefixed elements : ['LOW-45', 'HIGH-53', 'HIGH-76', 'HIGH-86', 'LOW-3', 'LOW-49']

Time complexity: O(n), where n is the length of the test_list.
Auxiliary space: O(n), where n is the length of the test_list.

Method #4: Using a for loop and ternary operator

This code iterates through the elements of test_list and appends either pref_1 or pref_2 to the element depending on the value of the element. The result is stored in the list res, which is then printed.

test_list = [45, 53, 76, 86, 3, 49]
pref_1 = "LOW-"
pref_2 = "HIGH-"
res = []

for x in test_list:
    res.append(pref_2 + str(x) if x >= 50 else pref_1 + str(x))

print("The prefixed elements : " + str(res))

The prefixed elements : ['LOW-45', 'HIGH-53', 'HIGH-76', 'HIGH-86', 'LOW-3', 'LOW-49']

Time complexity: O(n), where n is the number of elements in test_list. 
Auxiliary space: O(n), where n is the number of elements in test_list.

Method #5: Using a dictionary

This uses the fact that True and False can be used as keys in a dictionary and maps the True key to the "HIGH-" prefix and the False key to the "LOW-" prefix. The prefix_map[ele >= 50] expression returns the appropriate prefix based on the condition.

test_list = [45, 53, 76, 86, 3, 49]
prefix_map = {True: "HIGH-", False: "LOW-"}
res = [prefix_map[ele >= 50] + str(ele) for ele in test_list]
print("The prefixed elements : " + str(res))

The prefixed elements : ['LOW-45', 'HIGH-53', 'HIGH-76', 'HIGH-86', 'LOW-3', 'LOW-49']

Time complexity: O(n), where n is the length of the test_list. 
Auxiliary space: O(n), as the res list will have the same length as the test_list.

Method 6:  By adding conditional prefix to the elements in a list is-

Use the reduce() function to iterate over the elements of the list and apply the conditional check to add the prefix accordingly. We initialize an empty list as the initial value of the accumulator and append the result of each iteration to it. Finally, we get the desired result.

# Python3 code to demonstrate working of 
# Conditional Prefix in List
# Using reduce function

from functools import reduce

# initializing list
test_list = [45, 53, 76, 86, 3, 49]

# printing original list
print("The original list : " + str(test_list))

# initializing pref 1
pref_1 = "LOW-"

# initializing pref 2
pref_2 = "HIGH-"

# solution encapsulated as one-liner and conditional checks using reduce() function
res = reduce(lambda acc, ele: acc + [pref_2 + str(ele) if ele >= 50 else pref_1 + str(ele)], test_list, [])

# printing result 
print("The prefixed elements : " + str(res))

The original list : [45, 53, 76, 86, 3, 49]
The prefixed elements : ['LOW-45', 'HIGH-53', 'HIGH-76', 'HIGH-86', 'LOW-3', 'LOW-49']

The time complexity of this program is O(n), where n is the length of the input list, as we are iterating over each element of the list only once using the reduce() function.

The space complexity of this program is also O(n), as we are creating a new list containing the prefixed elements in the reduce() function. 

manjeet_04

Follow

Improve

Article Tags :

• Python
• Python Programs
• Python list-programs
• Python string-programs

Practice Tags :

• python