Python str型とint型が混ざった標準入力の扱い方

Question

Pythonのintとstrが混ざった標準入力の扱いについての質問です。

標準入力として
N(生徒数)
t(文系/理系), e(英語), m(数学), s(理科)

として、

3
bunkei 90 70 40

のように数値が入力されるとします。

文系は3教科計210点かつ英語80点
理系は3教科計210点かつ数学80点
で合格し、最後に合格者数を出力するプログラムを書いています。

tをstr型、e, m, sをint型として扱いたいのですが、
力技で下記のように書くしか思いつきませんでした。

この場合標準入力をよりうまく扱う方法はあるでしょうか？
教えていただければと思います。

#生徒数
N = int(input())

#合格者数
accept = 0

#文系、理系それぞれに点数の基準を設け、合格ならacceptに1を加える
for i in range(N):
    t, e, m, s = input().split()
    if t == "bunkei" and int(e)+int(m)+int(s) >= 210 and int(e) >= 80:
        accept += 1
    elif t == "rikei" and int(e)+int(m)+int(s) >= 210 and int(m) >= 80:
        accept += 1

print(accept)

Answer 1

あんまり奇麗じゃないですけど、比較的ましな方法です。

t, *scores = input().split()
e, m, s = map(int, scores)

Comments

[Justin04]

すみません、いつもありがとうございます。
scoreに一旦入れてからint型に変換すれば良いのですね！

Answer 2

９行目を、

t, e, m, s = [int(v) if i > 0 else v for i,v in enumerate(input().split())]

print(*(map(lambda x : x * 4, [t, e, m, s])))
'''
入力例
5 6 7 8
出力
5555 24 28 32
'''

N = int(input())
l = [[int(v) if i > 0 else v for i,v in enumerate(input().split())] for h in range(N)]
acpt = 0
for t,e,m,s in l:
    if e + m + s >= 210:
        if (t == "bunkei" and e >= 80) or (t == "rikei" and m >= 80):
            acpt += 1
print(acpt)

Answer 3

コードは長くなってますが、合格判定を関数に分離してみた例を示します。
入力データの解析 (文字列のままにする項目、数字にする項目の処理もそこで行っています)

def hantei(info):
   kind = info[0]
   scores = [int(x) for x in info[1::]]

   total = sum(scores)
   if kind == "bunkei":
      key_score = scores[0] # 英語
   elif kind == 'rikei':
      key_score = scores[1] # 数学

   return total >= 210 and key_score >= 80


N = int(input())
infos = []
for i in range(N):
   infos.append(input().split())

hanteis = [hantei(x) for x in infos]
print(hanteis.count(True))

Answer 4

一旦その行全体を文字列として取り込みます
それをスペース区切りで分解して、あとは文字列で使うなり整数変換して使うなりすればよろしい

Comments

[LouiS0616]

質問のコードの意味分かってますか？