전치 암호

# 전치 암호
import math

text = input("문자열 입력 : ")
text = text.upper()
print(f"\n 문자열 : {text} \n")
key = str(input("키값 입력 : "))
print(f"\n 키값 : {key} \n")

text_arr = [ [""] * len(key) for i in range( math.ceil( len(text) / len(key) ) ) ]
result_arr = [ [""] * len(key) for i in range( math.ceil( len(text) / len(key) ) ) ]

#입력 문자열 표 변환
idx = 0
for row in range(0, len(text_arr)) :
    for col in range(0, len(text_arr[row])) :
        if ( len(text) > idx ) :
            text_arr[row][col] = text[idx]
        else :
            text_arr[row][col] = "Z"
        idx += 1

# 입력 문자열 표 출력
for row in range(0, len(text_arr)) :
    for col in range(0, len(text_arr[row])) :
        print(text_arr[row][col], end=" ");
    print("")
print("")

# 암호화 변환
for row in range(0, len(result_arr)) :
    for col in range(0, len(result_arr[row])) :
        result_arr[row][col] = text_arr[row][int(key[col]) - 1]

# 암호화 문자 표 출력
for row in range(0, len(text_arr)) :
    for col in range(0, len(text_arr[row])) :
        print(result_arr[row][col], end=" ");
    print("")
print("")

# 최종 암호문 열 순서로 출력
row = 0
print(" 암호문 : ", end="")
for col in range(0, len(text_arr[row])) :
    for row in range(0, len(text_arr)) :
        print(result_arr[row][col], end="");
print("")