Deszyfrowanie szyfru kolumnowego

def encrypt(message, key):
    # wyznaczamy liczbę kolumn
    num_cols = len(key)
    # dzielimy długość słowa przez liczbę kolumn i określamy resztę z dzielenia
    num_rows, remainder = divmod(len(message), num_cols)
    # jeżeli reszta z dzielenia jest różna od 0, to dodajemy 1 do liczby wierszy
    if remainder != 0:
        num_rows += 1
    # tworzymy pustą tablicę do przechowywania słowa
    table = [[''] * num_cols for i in range(num_rows)]
    # wypełniamy tablicę poziomo, w kolejnych kolumnach
    for i, char in enumerate(message):
        row, col = divmod(i, num_cols)
        table[row][col] = char
    # szyfrujemy przez odczytanie kolumn w kolejności określonej przez klucz
    ciphertext = ''
    for col in key:
        ciphertext += ''.join(table[row][col] for row in range(num_rows))
    return ciphertext