pythonの外部入力を自動化したい

Question

▪️ ご質問内容
pythonでinput関数の外部入力を自動化したいです。
どのような手段がありますのでしょうか？
現在は、シェルスクリプトを用いて外部入力をしようと試みているのですがうまくできません。
何か方法がありましたら教えて欲しいです。
OSはMacを用いています。

python test.py
# ここで”a b”と入力させるようにしたい

   test = input()
   print(test) # a b

▪️ 質問の背景
競技プログラミングAtcoderを始めたのですが、
問題の入出力例をいちいちキーボードで入力するのが手間であるため、簡易化したいと考えています。

ネットで調べても簡単にできる良い方法が見つからなかったためご教授いただきたいです。
よろしくお願いいたします。

Answer 1

標準入出力のリダイレクトが使えます。入力のデータをファイルにあらかじめ保存しておき、それをキーボード入力から差し替えます。例えば4行からなる整数を足して合計を出力する以下のようなスクリプトがあったとして、

result = 0
for i in range(4):
    v = int(input().rstrip())
    result += v

print(result)

あらかじめ以下のような内容のファイルを用意します。ファイル名はinput.txtとします。

1
2
3
4

これを、python3の実行時にコマンドラインから < input.txtとして指定します。すると、キーボードからの入力がinput.txtファイルのからのものに切り替わります。実行結果がすぐに出ます。

$ cat input.txt
1
2
3
4

$ python3 test.py < input.txt
10

詳しくは「標準入出力 リダイレクト」などで検索してみてください。

ちなみに > output.txt とすると出力がファイルoutput.txtに保存されます。これを応用すると競プロのテストケースが分かっているような場合、自動で回答と突き合せることができるようになるということです。簡易オレオレ答え合わせシステムが作れる、と言うことですね。標準入出力およびそのリダイレクトは、Windowsのコマンドプロンプトでも使えます。

Comments

[Msk07]

こんな方法があるんですね！
ありがとうございます！！
勉強になりました！

[dodox86]

標準入出力はUN*X、Windows等の主なOSでサポートされる機能であり、Python以外のプログラミング言語でもたびたび出てくる概念で便利ですので、覚えておくと良いと思います。

[Msk07]

はい、調べて見たところ、Python以外にも使える内容とのことで
大変勉強になりました。ありがとうございます！

Answer 2

sh スクリプト中に 入力データを記載しておく方法、
外部ファイルに入力データを記載しておき、sh スクリプトを実行するときの引数にそのファイルを指定する方法
を示します。

p.sh

python3 p.py << EOS
a b
EOS

for arg in "$@"; do
    echo "file:". ${arg}
    python3 p.py < ${arg}
done

p.sh

line = input()
print("line:", line)

実行例

参考情報

• bashのヒアドキュメントを活用する

https://qiita.com/take4s5i/items/e207cee4fb04385a9952