Skip to content
Related Articles

Related Articles

Improve Article
Python | Merge first and last elements separately in a list
  • Last Updated : 07 Jun, 2019

Given a list of lists, where each sublist consists of only two elements, write a Python program to merge the first and last element of each sublist separately and finally, output a list of two sub-lists, one containing all first elements and other containing all last elements.

Examples:

Input : [['x', 'y'], ['a', 'b'], ['m', 'n']]
Output : [['x', 'a', 'm'], ['y', 'b', 'n']]

Input : [[1, 2], [3, 4], [5, 6], [7, 8]]
Output : [[1, 3, 5, 7], [2, 4, 6, 8]]

 
Approach #1 : List comprehension and zip




# Python3 program to Merge first and last
# elements separately in a list of lists
  
def merge(lst):
      
    return [list(ele) for ele in list(zip(*lst))]
      
# Driver code
lst = [['x', 'y'], ['a', 'b'], ['m', 'n']]
print(merge(lst))
Output:
[['x', 'a', 'm'], ['y', 'b', 'n']]

 
Approach #2 : Using Numpy array



First convert the given list to numpy array and then return transpose of the array, and finally convert the array to list.




# Python3 program to Merge first and last
# elements separately in a list of lists
import numpy as np
  
def merge(lst):
    arr = np.array(lst)
    return arr.T.tolist()
      
# Driver code
lst = [['x', 'y'], ['a', 'b'], ['m', 'n']]
print(merge(lst))
Output:
[['x', 'a', 'm'], ['y', 'b', 'n']]

 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 :