PRACTICAL 3: SIMPLE COLUMNAR TRANSPOSITION

CODE:
message = "WE ARE DISCOVERED SAVE YOURSELF"
key = "ZEBRAS"
n = len(key)

text = message.replace(" ", "")
while len(text) % n != 0:
    text += "X"

encrypted = ""
for letter in sorted(key):
    col = key.index(letter)
    for i in range(col, len(text), n):
        encrypted += text[i]
print("Encrypted:", encrypted)

decrypted = [""] * len(text)
k = 0
for letter in sorted(key):
    col = key.index(letter)
    for i in range(col, len(text), n):
        decrypted[i] = encrypted[k]
        k += 1
print("Decrypted:", "".join(decrypted).rstrip("X"))
