if/elifの条件が似ているのでコードを修正し短くすっきりさせたい

Question

前提・実現したいこと

多重ループの中にあるwhileに、if/elif/else条件が設定されています。
ifとelifの条件は少し異なるものの、大部分が同じ内容です。
統合させコードをすっきりさせたいです。アドバイスをいただけたら大変助かります。

location = input()　という文を入れていまして、
locationに与えられた文字列が「HOME」だったらif、「OFFICE」だったらelifに移ります。それ以外はelse。if/elif内では最初に「HOME」「OFFICE」の表記を「Home」「Office」に直す文を与えています。しかしながら、そのあとに続く命令はif/elifどちらも全く同じです。

該当のソースコード

以下、抜粋になります。

    inputCheck = True
    while inputCheck:
      location = input("Location: ")
      location = location.upper()
      if len(location) == 0:
        print("No data.")
        inputCheck = False

      elif len(location) > 0:    
        checkLocation = True
        while checkLocation:

#以下のif/elifが修正したい部分になります。

          if location == "HOME" :
            location = "Home"
            listing = location + "."
            listings.append(listing)
            output = " It is at " + location + "."
            print(output)
            checkLocation = False   
            inputCheck = False
          
          elif location == "OFFICE":
            location = "Office"
            listing = location + "."
            listings.append(listing)
            output =  " It is at " + location + "."
            print(output)
            checkLocation = False   
            inputCheck = False
      
          else:
            checkLocation = False    

試したこと

追加でwhile(listPrint)を作って試してみましたが、うまくいきませんでした。
挿入する場所が間違っているのでしょうか。

    inputCheck = True
    while inputCheck:
      location = input("Location: ")
      location = location.upper()
      if len(location) == 0:
        print("No data.")
        inputCheck = False

      elif len(location) > 0:    
        checkLocation = True
        while checkLocation:

#if/elif文を最小限に修正し・・・

          if location == "HOME" :
            location = "Home"
            listPrint = True
          
          elif location == "OFFICE":
            location = "Office"
            listPrint = True
            
          else:
            checkLocation = False

#ここにlistPrintを追加しました。
            
        listPrint = False
        while listPrint == True:
              listing = location + "."
              listings.append(listing)
              output = " It is at " + location + "."
              print(output)
              listPrint = False
              checkLocation = False   
              inputCheck = False      
 

補足情報（FW/ツールのバージョンなど）

python3.8.2

Comments

[退会済みユーザー]

Pythonらしいコードを俗にパイソニックと言いますが、if location == X, elif location == Y, ... と言う感じで同じ変数をチェックしている場合はdictに置き換えるとパイソニックです。また入力がされたかどうかもif len(location) == 0とせずにif not locationとするだけでlocationがNoneまたは空文字列であることがチェックできます。これもまたパイソニックな書き方です。

[pystudent]

そうなんですね。まだPython勉強し始めたばかりで知りませんでした。ご丁寧に教えていただきまして大変ありがとうございます。さっそく自分のコードに適用させていただきたいと思いますm(__)m

Answer 1

以下の点を変更してみてはいかがでしょうか。

• in を使って、locationが"HOME"や"OFFICE"などのいずれかにマッチするかを確認
• str.capitalize() でlocationの先頭のみ大文字に
• f-string を使って文字列の組み立て
• checkLocation = False は共通なのでifブロックの外に出す(これでelse節をカット)

変更後のイメージは、こんな感じになります。

target_locations = ("HOME", "OFFICE")
listings = []

location = "OFFICE"

if location in target_locations:
    listings.append(f"{location.capitalize()}.")
    print(f" It is at {location.capitalize()}.")
    inputCheck = False
checkLocation = False

Comments

[pystudent]

ご回答いただきまして大変ありがとうございました。知りたかったcapitalize機能とそのほかにも+αのアドバイスもいただけ、とても参考になりました。参考例もありがとうございました。

Answer 2

While や check*** 変数の処理がわかりにくかったので全体を書き換えてみました。
p.py

LOCATIONS = {"HOME": "Home.", "OFFICE": "Office."}

listings = []
while True:
    location = input("Location: ")
    if len(location) == 0:
        break;

    val = LOCATIONS.get(location.upper())
    if val:
        listings.append(val)
        print(" It is at " + val)

print("End of Data.")
print(listings)

実行例：

空行をいれると、処理が終了します。
Home や office (大文字・小文字は区別しない) を入力すると、 listings にデータを追加し、
"it is ag ***." のような出力をします。
それ以外の入力は何もしません。

入力が終了したら、 "End of Data." を出力し、さらに listings[] の値も表示します。

Comments

[pystudent]

ご回答いただきましてありがとうございました。dictionary機能は使用したことがありませんでしたので、大変参考になりました。実際にコードの実行までしていただきとても感謝しております。本当にありがとうございました。また、アカウントフォローさせていただきました。

Answer 3

文字列の文字を最初の文字のみ大文字、他を小文字に変化するにはstr.capitalize()が使えます。

https://docs.python.org/ja/3/library/stdtypes.html#str.capitalize

ですので

location = "Home" # or "Office"

の部分を

location = location.capitalize()

に置き換えることで、２つの条件文の中を完全に同じ内容にできるのではないでしょうか。

        #以下のif/elifが修正したい部分になります。
        if location in ["HOME", "OFFICE"]:
            location = location.capitalize()
            listing = location + "."
            listings.append(listing)
            output = " It is at " + location + "."
            print(output)
            checkLocation = False
            inputCheck = False

        else:
            checkLocation = False

Comments

[pystudent]

的確なアドバイス大変ありがとうございました。教えてくださったcapitalize()機能を使わせていただきたいと思います。説明のリンクもつけていただき、本当に為になりました。ありがとうございます。

Answer 4

比較する文字列と代入する文字列をテーブルに入れといて、ループさせれば、コードを書くのは一回分で済みますね

Comments

[pystudent]

ご回答いただきましてありがとうございました。ループ機能を使用してコードを修正したいと思います。

Answer 5

if location in ["HOME", "OFFICE"]:
    listing = location.title() + "."
    ...
    ...

obj in (list likeなobj)で真偽判定できます。

Comments

[pystudent]

ご回答いただきまして大変ありがとうございました。教えていただきましたobj in []機能を使用して、コードを修正させていただきたいと思います。