python3 標準入力 まとめ


はじめに

python3 標準入力まとめです。
paizaの問題などでよく使いそうなものをまとめています。

1行に要素1個

単語1個

入力
aiueo
main.py
hoge = input()

print(type(hoge))
print(hoge)
出力
<class 'str'>
aiueo

整数1個

入力
100
main.py
hoge = int(input())

print(type(hoge))
print(hoge)
出力
<class 'int'>
100

1行に要素n個

1行に複数要素ある場合。
要素間はスペース区切りを想定しています。

単語n個を配列に格納

入力
foo bar baz
main.py
list = [s for s in input().split()]

print(type(list))
print(list)
出力
<class 'list'>
['foo', 'bar', 'baz']

単語n個を各変数に格納

入力
foo bar baz
main.py
str1, str2, str3 = [s for s in input().split()]

print(type(str1))
print(str1)

print(type(str2))
print(str2)

print(type(str3))
print(str3)
出力
<class 'str'>
foo
<class 'str'>
bar
<class 'str'>
fbaz

整数n個を配列に格納

入力
100 200 300
main.py
list = [int(i) for i in input().split()]

print(type(list))
print(list)
出力
<class 'list'>
['100', '200', '300']

整数n個を各変数に格納

入力
100 200 300
main.py
num1, num2, num3 = [int(i) for i in input().split()]

print(type(num1))
print(num1)

print(type(num2))
print(num2)

print(type(num3))
print(num3)

出力
<class 'int'>
100
<class 'int'>
200
<class 'int'>
300

n行に要素m個

n行に整数m個

要素5つ, 3行の入力を想定しています。

入力
5 3
10 11 12 13 14
20 21 22 23 24
30 31 32 33 34
main.py
n, m = [int(i) for i in input().split()]
table = [[int(i) for i in input().split()] for m in range(m)]

print(type(n))
print(n)
print(type(m))
print(m)

print(type(table))
print(table)
出力
<class 'int'>
5
<class 'int'>
3
<class 'list'>
[[10, 11, 12, 13, 14], [20, 21, 22, 23, 24], [30, 31, 32, 33, 34]]

おわりに

しっかりしたエラー処理などはしていないため、使用の際は気をつけてください。。。

参考文献