0と1の数字の並びを次の規則に従って変化させるようなプログラムはかけるか。

Question

入力ファイルdata.datは次のように、0と1が並んでいます。

1 1 1 1 0 0 0 0 1 1 1 1 1 1 1 0 1 1 1 0 0 0 1 1 1 1 1 0 0 0 1 1 0 1 1 0 1 1 1 1 1 1 1 0 0 0 1 1

まず、すべての11011という並びを11111に変更します。

1 1 1 1 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 0 0 0 1 1 1 1 1 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 0 0 0 1 1

つぎに任意の0からつぎの0までが10字以内であれば、その0から次の0までの1を全て0に変えます。

1 1 1 1 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 0 0 0 1 1

このようなプログラムをpythonで書きたいのですが、どのようにすればよいでしょうか。なお、スペースを無駄にしないように横に書きましたが、data.datは
1
1
1
1
0
0
…
のように一文字ずつ改行されているものとします。想定としてはdata.datは5万行程度のデータなので何かしらの工夫が必要だと思うのですが…。

追記１

たとえば、一つ目のプロセスに関しては、

lines = len(open('data.dat').readlines())

for p in range(1,lines-5):
	target_line1 = linecache.getline("data.dat",p)
	target_line2 = linecache.getline("data.dat",p+1)
	target_line3 = linecache.getline("data.dat",p+2)
	target_line4 = linecache.getline("data.dat",p+3)
	target_line5 = linecache.getline("data.dat",p+4)

	if target_line1 == 1 and target_line2 == 1 and target_line3 == 0 and target_line4 == 1 and target_line5 == 1:
　　　　#p+3行目を1に書き換える

とすればよいのでしょうが、data.datの特定の行を書き換える方法がわかりません。

Comments

[otn]

技術的に何が分かりませんか？逆に、何がわかりますか？

[coco_bauer]

改行を全て削除して連続した文字列に変換し、正規表現(reモジュール)を使って置換すれば簡単そうに思えます。

Answer 1

意外と面倒ですね、これ。

import re

#
# read data
with open('data.dat') as f:
    src = f.read().replace('\n', '')

print(src)

#
# replace
while '11011' in src:
    src = src.replace('11011', '11111')

print(src)

#
# replace
src = re.sub(
    r'01{1,10}0',
    lambda m: '0' * (m.end() - m.start()),
    src
)
print(src)

実行結果 Wandbox

111100001111111011100011111000110110111111100011
111100001111111111100011111000111111111111100011
111100001111111111100000000000111111111111100011

一段階ずつじっくり考えていけば分かります。
今後も同様の処理を書くのでしたら、特に正規表現の扱いに慣れておいてください。

Comments

[physics303]

ありがとうございます。
正規表現については勉強してみます（手元の入門書にはかいてませんでした）。

[physics303]

改行して、出力したいのですが、
src = str(src)
X = src.replace("1","1\n")
Y = X.replace("0","0\n")
print(Y)
とすればいいですね。
もっとスマートな方法はありますか？

[LouiS0616]

一番わかりやすいのは一文字ずつprintする方法です。
スマートな方法だとstr.joinを使ったりとか。