HackerRank pythonのList問題

Question

前提

2
Harsh 25 26.5 28
Anurag 26 28 30
Harsh

結果　26.5

という問題がある。

これは、n(2)を入力して、名前とスコアを書いていき、query_nameと名前が一致した時に、スコアの平均を小数第二位まで表示する問題です。以下がそのソースコードです。

該当のソースコード

if name == 'main':
n = int(input())
student_marks = {}
for _ in range(n):
name, *line = input().split()
scores = list(map(float, line))
student_marks[name] = scores
query_name = input()
sum = 0
if(name == query_name):
for i in range(len(scores)):
sum += scores[i]
print(f'{sum/len(scores):.02f}')

###　正答の問題

###　誤答の問題

###　分からないこと
正答と誤答の違いが分からないです。

Comments

[west_urad]

原因がわかりました

Answer 1

問題解決したようなので、別のアドバイスを。
自分でsum計算せずに、sum関数を使うといいですよ。
名前の存在確認は in 演算子を使うといいです。
コードは「コードの挿入」(</>)を使って書くとインデントされます。

students = {}
for _ in range(int(input())):
    name, *scores = input().split()
    students[name] = list(map(float, scores))
name = input()
if name in students:
    scores = students[name]
    print(f'{sum(scores)/len(scores):.02f}')

Comments

[west_urad]

丁寧にありがとうございます。

Answer 2

インデントがぐちゃぐちゃなのではっきりしたことは言えませんが、
平均を求めるときに参照している変数がscoresのため常にリスト最後の人の平均を求める形になっていますよ。
if(name == query_name):をしているので求めるのがリスト最後の人でなければ平均の計算自体やっていないかもしれませんね。
現に画像の結果を見る限りsuccessになっているのは平均を求める必要があるのがリスト最後の人の時だけですね。
せっかくstudent_marks[name]で連想配列に収めているのですから、student_marks[query_name]の平均を求めてみては。

Comments

[west_urad]

考えすぎてました。
query_nameの後に

q = student_marks[query_name]
sum = 0.0

for i in range(len(q)):
sum += q[i]

print('{:.02f}'.format(sum/3))
と入力したら解決しました。