python3の標準入力

研究のデータ処理がめんどくさくてなんとか自動で処理できないものかと思い、最近python3を触り始めました。ただほんとに初心者でやろうと実装できるわけではないので、ひとまず慣れるためにAtCoderのBeginner問題を練習で解いているところです。今回はプログラムを動かすにあたっての第一歩、値の入力について個人的メモもかねて書きます。

• 値の入力

N = input()

入力した値をそのままNに代入します。整数型で入力したい場合は、

N = int(input())

となります。他の型も同様にintの部分を対応する表現に変えたら大丈夫です。

• 1行でスペース区切りの複数入力

a, b, c = input().split()

a, b, cのそれぞれに入力した順で値が代入されます。全てを整数型にしたい場合は、

a, b, c = map(int, input().split())

と書くとできます。入力した値をlist化したい場合は、

l = list(map(int, input().split()))

もしくは

l = [int(i) for i in input().split()]

と書けばできます。

• 複数行(N行)の値の入力

今回は各行に1つの値しかない場合のみ書きます。複数行に複数の値がある場合はまた今度。

a = [int(input()) for i in range(N)]

とりあえず最初の段階としてはこれぐらいが扱えれば大丈夫なのかな、と感じました。

参考記事は、

Pythonで競技プログラミングする時に知っておきたいtips(入出力編) - Qiita

です。