def pad_to_square(a, pad_value=0):
    m = a.reshape((a.shape[0], -1))
padded = pad_value * np.ones(2 * [max(m.shape)], dtype=m.dtype)
Создание квадрата матрицы и заполнение его желаемым значением в numpy