input()の内容によって場合分けをしたい

Question

python初学者の者です。
以下のような課題に取り組んでいるのですが、悩んでいます。

標準入力を使って生徒の成績を管理するツールを作ってください。

“save 生徒名 点数“とすると、その生徒の点数を保存。
“save 生徒名 点数“とした場合、
生徒名がすでに保存されている時は、点数を上書きするか警告文を出して、yes/noで上書きの選択を行う。
“get 生徒名“とするとその生徒の点数を表示。
“get 生徒名“を行った場合、生徒が登録されていない場合はErrorと出力させて処理を継続
“quit” とするとプログラムが終了する。
“average” すでに保存されいる生徒全員の平均点
“登録されている生徒XX人の平均点はXX.XX点です”

averageとquitの部分はできたのですが、saveとgetの部分をどう書けばいいのか分かりません。
生徒名や点数は多くのパターンがあると思うのですが、それをどう書けばいいのかが分かりません。

どなたか教えていただければ幸いです。

Comments

[meg_]

> averageとquitの部分はできたのですが
出来たコードを掲載されると回答しやすくなるかと思います。

Answer 1

まず、プログラミングっていうのは『こう書けばいい』というものではありません。

プログラミングは『現実世界のシミュレーション』です。

なので、とりあえず、プログラミングでは…とかPythonでは…とかは置いておきましょう。
先に『手作業でやるなら』と考えてみましょう。数学の問題を解くとか、毎日やる作業とかと考えてみるのです。

そうすると、ある意味、

ホワイトボードに”save 生徒名 点数” のように、コマンドのように何行にも連なった処理列がある。あなたはその処理リストを読んで、それらの処理をせよ。

[処理リストの例]
save Tanaka 100
save Yamada 80
save Oosaka 100
…
get Yamada
average
quit 

といった感じでしょうか。

そういう場合、どのようにしますか？

私なら、『規則性』を見つけます。

save の構文は常に、

save <人名> <点数>

となっています。

それと対になっているであろう、get は、

get <人名>

となっていて、quitやaverageは

quit

のように一単語になっていますね。

save の構文を再度見ると、

<命令> <人名> <点数>

のようになっていますね。

つまり、『命令と人名等は「半角スペースによって区切られている」』ですね。

ということは、単純に半角スペースを区切り文字にして読んでいけばいいです。

getも同様にすると、単語数は違えど、半角スペースで区切られています。

quitやaverageは一単語のみで、半角スペースはありませんが、『半角スペースまで、または最後の文字までを一単語とする』と見なせば、条件を満たします。

よって、一行あたりのコマンドは『半角スペースまで、または最後の文字までを一単語とする』と考えることができます。

それでは、ホワイトボードに書かれた処理をやっていきましょう。

[処理リスト]

save Tanaka 100
save Yamada 80
save Oosaka 100
…
get Yamada
average
quit 

上記処理リストの最初は、saveですね。

これは規則性から、『Tanakaさんの点数は100点として記録する』ですね。

これはホワイトボードの余白に、

Tanaka: 100点

とかみたいに書いていきます。

で、二番目の処理もsaveですね。
なので、

Tanaka: 100点
Yamada: 80点

のように追記していく。

三行目以降のsaveも同様にやっていく。

get Yamada のやつは、余白に書いたアレから、Yamada のものを探して、それを結果用のシートかなんかに記述。

average は『平均点を表示』ですから、平均点を算出する。
その都度計算してもいいですが、これが、
average, average, average, average…
のように複数回だとどうなるでしょうか。
計算するのもめんどくさいですね。
流石に同じことを延々に聞くことはありませんが、登録 → 現在の平均点 をループすることはあり得ますね。

毎回やるのは面倒です。

で、よくよく考えると、

平均 = 合計点数 ÷ 人数

ですね。この合計点数は、save時でも計算ができます。
単純に現在の値に足せばいいだけなので。

その合計点数を人数で割ればいいだけです。

quitは単純に処理を終了すればいいだけですし。

これを実装するだけです。

『半角スペースを区切り文字にして…』の読み込みは、『半角スペースを区切り文字として、文字列分割をする』だけでいいです。
そうすれば、saveとかのコマンドだけ取り出すことも可能です。

なので、二次元リストでやればいけそうですね。

Cmd[0][0] とかで取り出せばいいだけですね。

saveの場合やgetの場合とかのように個数が違う場合もあります。
なので、

saveの場合： 1番目と2番目を使う
getの場合: 1番目のみ使う
averageやquitの場合: リスト内のデータは使わない

的な感じで、if文で篩い分けするときに考えればいいだけです。

そして、デバッグまではする。
これが基本です。

Answer 2

標準入力をリストとして受け取り、その第一要素別に処理を行い、辞書型でデータの管理を行えば良いのではないでしょうか。

dic = {}

while True:
	li = list(map(str, input().split()))

	if li[0] == 'save':
		if li[1] not in dic:
			dic[li[1]] = int(li[2])
		else:
			print('上書きをしますか? yes/no')
			ans = str(input())
			if ans == 'yes':
				dic[li[1]] = int(li[2])

	elif li[0] == 'get':
		if li[1] not in dic:
			print('Error')
		else:
			print(dic[li[1]])

	elif li[0] == 'quit':
		exit()

	elif li[0] == 'average':
		avg = round(sum(list(dic.values())) / len(dic), 2)
		print('登録されている生徒' + str(len(dic)) + '人の平均点は' + str(avg) + '点です')

Answer 3

「ファイルに書き出す」等の要求が無いのであれば、
メモリ内で辞書型に持たせてしまうのが手っ取り早いと思われます。

以下、エラー処理なしでかなり適当に書いたサンプルです。

data = {}
while True:
    p = input("command >>>")
    plist = p.split(' ')
    if plist[0] == 'save':
        if data.get(plist[1]):
            p = input("{} is exists. update? yes/no >".format(plist[1]))
            if not p=='yes':
                continue
        data[plist[1]] = int(plist[2])
    elif plist[0] == 'get':
        if data.get(plist[1]):
            print("{} is {} points.".format(plist[1], data[plist[1]] ))
        else:
            print("{} is not exists.".format(plist[1]))
    elif plist[0] == 'quit':
        break
    elif plist[0] == 'average':
        print("登録されている生徒{}人の平均点は{}点です".format(len(data), sum(data.values())/len(data) ))
    else:
        pass