Python｜for文で出力した内容をテキストとして変数に代入したい

Question

実現したいこと
・for文で出力した内容を変数に代入したい

#例
for num in range(5):
    print(num)

#出力
1
2
3
4
5

例えばこのようにfor文で1〜5までの数字が出力されますが、全体（改行含む）をテキストとして変数に代入する方法が分かりましたら、ご教授いただけたら幸いです。

Answer 1

こんにちは。

ご質問で望まれているのは、おそらくKSwordOfHasteさんのご回答であるとは思いますが、
質問の趣旨を

「printで出力していった内容を、文字列として変数に取り出すにはどうしたらいいか？」

というものと解釈すると、以下が回答の例になります。

import sys
import io

# 文字列IOストリームを初期化して、f に代入
with io.StringIO() as f:

    # 標準出力を f に切り替える。
    sys.stdout = f

    # fに切り替わった標準出力に対してrange(5)の要素を、printで出力
    for num in range(5):
        print(num)

    # f に出力されたものを文字列として取得
    text = f.getvalue()

    # 標準出力をデフォルトに戻して text を表示
    sys.stdout = sys.__stdout__
    print("START\n" + text + "END")

上記のコードを（python 3.4.3で）実行すると、以下

START

0
1
2
3
4
END

のように出力されました。この出力結果から、上記のコード中

    for num in range(5):
        print(num)

によって出力された内容が、

text = f.getvalue()

によって、 text に文字列として取り出せていることが分かります。

以上、参考になれば幸いです。

参考：io.StringIO

Comments

[mrgishi]

ありがとうございます！
上記で変数に代入できました！今後も使えそうなものでしたので、大変助かります！

[jun68ykt]

解決されたようで、よかったです！

Answer 2

joinやmapを組み合わせてこんなふうにも書けると思います。

lines = '\n'.join(map(str, range(5)))

なお各行の途中に改行文字(NL)が挟まる形になるので最終行の後ろにはNLがつきません。その点にはご注意を。

Answer 3

ご回答ありがとうございました！
jun68yktさんの回答で・for文で出力した内容を変数に代入したいを実現することができました。

import sys
import io

# 文字列IOストリームを初期化して、f に代入
with io.StringIO() as f:

    # 標準出力を f に切り替える。
    sys.stdout = f

    # fに切り替わった標準出力に対してrange(5)の要素を、printで出力
    for num in range(5):
        print(num)

    # f に出力されたものを文字列として取得
    text = f.getvalue()

    # 標準出力をデフォルトに戻して text を表示
    sys.stdout = sys.__stdout__
    print(text)