array([[1, 1, 2],
       [3, 3, 4],
       [3, 3, 4],
       [3, 3, 4]])
array([[0, 0, 0, 2, 0],
       [0, 0, 0, 0, 2],
       [5, 0, 0, 1, 1],
       [0, 5, 0, 1, 1],
       [0, 0, 5, 1, 1]])