Image Grid Montage

Problem

Tile a list of equal-shaped images into a rows x cols montage.

Images are placed row-major at (index // cols, index % cols). You may assume len(images) equals rows * cols.

case 1

in: [[[[1]],[[2]],[[3]],[[4]]],2,2]

expected: [[1,2],[3,4]]

case 2

in: [[[[1,2]],[[3,4]]],1,2]

expected: [[1,2,3,4]]

case 3

in: [[[[1,2],[3,4]],[[5,6],[7,8]],[[9,10],[11,12]],[[13,14],[15,16]]],2,2]

expected: [[1,2,5,6],[3,4,7,8],[9,10,13,14],[11,12,15,16]]

Starter codePython

def montage(images, rows, cols):
    # Your code here
    pass

3 test cases · run your code against them in the editor