改行済み文字列において一行ずつ区切り文字をつけたい

Question

下記のように、既に改行された文字列があり、

a.txt

one
two
three

b.txtのように[]のような区切り文字をつけたいと思っています。

b.txt

[one]
[two]
[three]

データの前処理に使う予定です。
お教えいただけると大変、助かります。
何卒、よろしくお願いいたします。

Comments

[otn]

前半と後半で全く違うことを言ってます。もういちどよく考えて質問を書き直して下さい。

Answer 1

with open('a.txt') as fin, open('b.txt', 'w') as fout:
    for line in ('[%s]' % s for s in map(str.strip, fin) if s):
        print(line, file=fout)

Comments

[hayataka2049]

これって、カンマ区切りで複数書けたんですね。勉強になりました

[YouheiSakurai]

です。ただ複数でencoding指定等をしだすと途端にPEP8に収まらなくなるので、contextlib.ExitStackも参照してやって下さい。

[退会済みユーザー]

ご回答ありがとうございました！
僕が、まさに、求めていた処理でした。
こんなに、短く書けるとは。

[hayataka2049]

ExitStackは大げさなので、行末に\置いて済ませることにします・・・（pep8でもその方法が紹介されてました）
動的に生成したいときに使うのだろう、くらいに思っておきます

[YouheiSakurai]

短くというよりもインデント階層を浅くしようとしました。少しトリッキーな部分もあるので、用法用量にはお気をつけください。

[YouheiSakurai]

hayataka2049さん、なるほど勉強になります。PEP8を読み返します。ありがとうございます。

Answer 2

in_filename = "a.txt"
out_filename = "b.txt"
with open(in_filename, 'r') as in_file, open(out_filename, 'w') as out_file:
    for line in in_file:
        out_file.write("[" + line.rstrip("\n") + "]\n")

参考情報

• “with” statement in Python to Open a file

http://cmdlinetips.com/2016/01/opening-a-file-in-python-using-with-statement/

...
Here is an example of using with statement in Python to open one file for reading and another file for writing.

with open(in_filename) as in_file, open(out_filename, 'w') as out_file:
for line in in_file:
...
...
out_file.write(parsed_line)
...

Comments

[退会済みユーザー]

ご回答ありがとうございました！

Answer 3

これができると何が幸せなのか、理解しかねますが・・・

lst = []
with open("a.txt", "r") as f:
    for line in f.readlines():  # 1行ずつ処理
        if line != "\n":  # 空の行は無視してみる
            lst.append(line)

with open("b.txt", "w") as f:
    for line in lst:
        f.write("[{}]\n".format(line.strip()))  # 1行ずつ書き込んでいく。stripは改行文字を消すため

Comments

[退会済みユーザー]

ご回答ありがとうございました！