python 空白を保持したまま文字列をリスト化してシーザー暗号を生成したい

Question

シーザー暗号を生成する関数を作っているのですがどうすれば空白を保持することができるでしょうか

例　to_encrypt("ab c",3) →　de f
to_encrypt("b c",-1) →　a b

コード
def to_encrypt(text,num):
newtext=""
li = [chr(i) for i in range(97, 97+26)]
text = text.replace(" ","")
textli = [i for i in text]
for i in range(len(textli)):
textli[i] = li[li.index(textli[i])+num]
for i in range(len(textli)):
newtext += textli[i]

return(newtext)

Comments

[hayataka2049]

コードは<code>ボタンで挿入できるコードブロックの中に入れてください

[hayataka2049]

コードは<code>ボタンで挿入できるコードブロックの中に入れてください

Answer 1

どうすれば空白を保持することができるでしょうか

見るからに怪しい行があるので、それを削除します。

text = text.replace(" ","")

なお、これを消しただけではまだエラーが出ます。

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "main.py", line 8, in to_encrypt
    textli[i] = li[li.index(textli[i])+num]
ValueError: ' ' is not in list

文字が空白であるときは置き換えをスキップすれば良いですね。

ただし

論理バグがあり、例えばzをまともに置き換えできません。
もうちょっと考えなおした方が良いでしょうね。

書いてみた

個人的な趣味全開です。

from string import ascii_lowercase as chars


def revolve(seq, shift):
    shift %= len(seq)
    return seq[shift:] + seq[:shift] 

def make_decode(shift):
    rep_table = str.maketrans(
        chars, revolve(chars, shift)
    )
    
    def decode(text):
        return text.translate(rep_table)
    
    return decode


decode = make_decode(3)
print(
    decode('ab c')
)

実行結果 Wandbox

de f

コードの書き方について

teratailには、コードを見やすく表示する機能があります。
質問編集画面を開き、コードを選択した状態で<code>ボタンを押してください。

特にPythonの場合、インデントが崩れるとコードの意味が変わってしまいます。

Comments

[Cespedes]

回答ありがとうございました
コードの書き方がよくわからず間違えてしまいました
ご指摘ありがとうございます

Answer 2

text = text.replace(" ","")をやめて、アルファベット26文字の中に入っていない文字は元の文字のままにする（26文字に含まれる文字だけ変換する）ような処理をすれば良いです。

def to_encrypt(text,num):
    newtext=""
    li = [chr(i) for i in range(97, 97+26)]
    textli = [i for i in text]
    for i in range(len(textli)):
        if textli[i] in li:
            textli[i] = li[li.index(textli[i])+num]            
    for i in range(len(textli)):
        newtext += textli[i]

    return(newtext)

print(to_encrypt("abcde f",-2))

Comments

[Cespedes]

回答ありがとうございました
無事解決しました！