Python - Custom element repetition

Given list of elements and required occurrence list, perform repetition of elements.

Input : test_list1 = [“Gfg”, “Best”], test_list2 = [4, 5]
Output : [‘Gfg’, ‘Gfg’, ‘Gfg’, ‘Gfg’, ‘Best’, ‘Best’, ‘Best’, ‘Best’, ‘Best’]
Explanation : Elements repeated by their occurrence number.

Input : test_list1 = [“Gfg”], test_list2 = [5]
Output : [‘Gfg’, ‘Gfg’, ‘Gfg’, ‘Gfg’, ‘Gfg’]
Explanation : Elements repeated by their occurrence number.

Method #1 : Using loop + extend()

The combination of above functions provide one of the ways in which this task can be performed. In this, we iterate using loop and perform extension of elements with repetition using extend().

Python3

# Python3 code to demonstrate working of  
# Custom elements repetition 
# Using  loop + extend() 
  
# initializing lists 
test_list1 = ["Gfg", "is", "Best"] 
test_list2 = [4, 3, 5] 
  
# printing original lists 
print("The original list 1 : " + str(test_list1)) 
print("The original list 2 : " + str(test_list2)) 
  
# using loop to perform iteration 
res = [] 
for idx in range(0, len(test_list1)): 
      
    # using extend to perform element repetition 
    res.extend([test_list1[idx]] * test_list2[idx]) 
      
# printing result  
print("The repeated list : " + str(res)) 

Output

The original list 1 : ['Gfg', 'is', 'Best']
The original list 2 : [4, 3, 5]
The repeated list : ['Gfg', 'Gfg', 'Gfg', 'Gfg', 'is', 'is', 'is', 'Best', 'Best', 'Best', 'Best', 'Best']

Method #2 : Using loop + zip()

This is yet another way in which this task can be performed. In this, we use zip() to match element with its repetition occurrence and perform required task of duplication.

Python3

# Python3 code to demonstrate working of  
# Custom elements repetition 
# Using  loop + zip() 
  
# initializing lists 
test_list1 = ["Gfg", "is", "Best"] 
test_list2 = [4, 3, 5] 
  
# printing original lists 
print("The original list 1 : " + str(test_list1)) 
print("The original list 2 : " + str(test_list2)) 
  
# using zip() to intervene elements and occurrence 
res = [] 
for ele, occ in zip(test_list1, test_list2): 
    res.extend([ele] * occ) 
      
# printing result  
print("The repeated list : " + str(res))

Output

The original list 1 : ['Gfg', 'is', 'Best']
The original list 2 : [4, 3, 5]
The repeated list : ['Gfg', 'Gfg', 'Gfg', 'Gfg', 'is', 'is', 'is', 'Best', 'Best', 'Best', 'Best', 'Best']

My Personal Notes

Python – Custom element repetition

Python3

Python3

Recommended Posts: