python3でinput()入力された複数行を取得したい

###前提・実現したいこと

paiza learning というものに取り組んでいます。
input_lines = input() に入力される、
複数行に渡る値を上手く取得できず、困っています。

input()に入力される複数行を上手く取得するにはどうすればいいのでしょうか。

input() には
20 10 10
5
25 10
10 23
20 15
70 70
10 30
と入力されることになっており、2行目の1桁の数字はそのあとに何行続くかを示しています(この場合、2行目の後にあと５行並ぶ)。

###試したこと

python3
1input_line = input()
2s = [input() for i in range(6)] 
3print(s)

とりあえずこんな感じで、リストに入れて値を取得しようとしてみました。

python3
1['5', '25 10', '10 23', '20 15', '70 70', '10 30']

するとこんな感じで出ました。
ここでinput1行目の 20 10 10 が　リストに入らないのはなぜでしょうか。
rangeの中の6は適当に入れただけで、本当はinput()に入力される2行目の1桁の数字分だけ行を取得したいです。

paiza learningの方はとっくに期限切れなので、これからの勉強のために複数行取得の方法が知りたいです!

行動規範の内容に同意します

回答3件

ベストアンサー

input()は、標準入力から1行受け取る組み込み関数です。1行目は最初のinput_line = input()で受け取られています。
【以下修正・追記済み】
おそらく1行目も問題を解くのに使うと思うので、これはそのまま受け取っておきましょう。split()を使ってリストにし、さらにmapなどを使ってリスト内のそれぞれの数値をint型に変換するという方法で、一気に[20, 10, 10]として取得できます。
この取得方法は競技プログラミングでよく使います。wakameさんが紹介されているページを参考にしたうえで、python3でのmapの使い方を学習されるとよいと思います。
（他にも、リスト内包を使う方法もあります。こちらの方がmapより読みやすいという方も多いので、学習してみてください）

2行目はもっと単純です。数値をひとつ受け取るだけなので、普通にinput()したものをint型に変換するだけです。

3行目以降は、for文などを使って1行目と同様の処理をN回繰り返せば、各行をそれぞれ[25, 10]などのように受取れます。

paizaのスキルチェックですので、具体的なコードの提示は控えますが、
競技プログラミング向けの標準入力を扱う方法はいろいろ情報がありますので、ぜひ調べてみてください。

投稿2018/01/18 12:37

編集2018/01/26 01:34

Udomomo

総合スコア1524

null0723

2018/01/18 13:01

input_line = int(input()) s = [input() for i in range(6)] print(s) 上記のようにしたところ、 Traceback (most recent call last): File "Main.py", line 5, in <module> input_lines = int(input()) ValueError: invalid literal for int() with base 10: '20 10 10' と出てしまいました... また、2行目の取得方法がよく分かりませんでした...

Udomomo

2018/01/18 13:29 編集

すみません、回答内の1行目の取得方法が正確ではなかったですね。修正しておきます。 2行目は単に数値をひとつ取得するだけです。こちらも追記しておきます。

null0723

2018/01/18 13:45

わざわざ丁寧にありがとうございます! 2行目を取得する方法はわかったのですが、１行目を取った後、どうやって２行目にいけばいいのでしょうか,,, 本当に初心者ですみません(´；ω；`)

Udomomo

2018/01/18 14:04

input()を1回行うと1行取得し、もう一度行うと次の行を取得します。なので、1行目の処理を書いた後に、2行目の処理を書くだけでよいです。3行目以降で「N回繰り返す」と書いているのも、3行目以降の入力がN行分あるからです。

null0723

2018/01/18 14:26

今やってみたらinput()を何個か羅列しただけでも一行ずつ取ってくれました!! なんこもinput書くと１行目が何個も出ると勝手に思ってました! ありがとうございました

行動規範の内容に同意します

入力部分ならば回答してもpaizaの規約には引っかからないだろうと考え、私ならこうするという意味で回答します。(当然他の回答者様の回答とモロ被りな部分があります。)
参考までにどうぞ。

python
1K,L,M = map(int,input().split())
2N = int(input())
3arr = [[int(i) for i in input().split()] for _ in range(N)]
4
5print('\nK: {},L: {},M: {}'.format(K,L,M))
6print('N:',N)
7for i,l in enumerate(arr):
8    print('a{}: {} b{}: {}'.format(i,l[0],i,l[1]))

入出力例

plain
120 10 10
25
325 10
410 23
520 15
670 70
710 30
8
9K: 20,L: 10,M: 10
10N: 5
11a0: 25 b0: 10
12a1: 10 b1: 23
13a2: 20 b2: 15
14a3: 70 b3: 70
15a4: 10 b4: 30