Merge multiple 2d lists considering axis in order

前端 未结 3 1779
予麋鹿
予麋鹿 2021-01-13 11:50

My purpose is to combine multiple 2d list in order such as:

a = [[1,2],[3,1]]
b= [[3,6],[2,9]]
c = [[5,1],[8,10]]
Expected: [[1,2,3,6,5,1],[3,1,2,9,8,10]]
         


        
3条回答
  •  情话喂你
    2021-01-13 12:20

    This might be another solution using numpy but this is much slower.

    import numpy as np
    
    a = [[1,2],[3,1]]
    b = [[3,6],[2,9]]
    c = [[5,1],[8,10]]
    
    print np.hstack((np.hstack((a,b)),c))
    
    # [[ 1  2  3  6  5  1]
    # [ 3  1  2  9  8 10]]
    

    and if you want it to have a list format then use

    np.hstack((np.hstack((a,b)),c)).tolist()
    

提交回复
热议问题