python３にて御助けください

Question

前提・実現したいこと

datefile

abcde（１）
11111＃１
11111＃２
11＃３

abcde（２）
111＃１

このようなデータがあるとき
最初にデータを読み込み、
if式でaが先頭に来たら、次の行を読みこみ、読み込んだ行（＃１）の文字数をカウント、次の行（＃２）の文字数もカウント、その次の行(＃３）で空白を認識したらストップというコードをpythonでできるだけシンプルに作りたいです。

(１）であれば１２、（２）であれば３といった感じです

試したこと

count = 0
for s in $stdin.read.split("\n") do
break if s.size == 0
count += s.size
end if gets[0] == "a"
puts count

rubyでこのように書きました
pythonで書く方法を教えていただきたいです
ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

python3です

ここにより詳細な情報を記載してください。

Answer 1

Pythonなど不要。そう、シェル芸ならね！

$ cat datafile|awk 'NR>1&&/^a/{print c;c=0}/^[^a]/{c+=length($1)}END{print c}'
12
3

Answer 2

メモリに充分余裕があるのであれば、こんなふうにすると楽かと。

• 文字列をごっそり読み込む
• aから始まる行で文字列を分割してリストにする
• リストの各要素について、空白文字を除去
• リストの各要素について、文字列の長さを数える

分割方法だけアドバイスします。
正規表現で、aから始まる行で分割すると楽です。

data = re.split('^a.+$', data, flags=re.MULTILINE)
print(data)

追記

よく考えたら、こんな方法もありますね。
こっちの方がメモリを圧迫しなくて良いかもしれません。

• 行数を格納するリストを用意する
• 一行一行読み取っていく

・ aから始まる行なら、リストに 0 をappendする
・ そうでないなら文字数をカウントし、リストの最終要素に足す

with open(ファイル名) as fin:
    char_counts = []
    for line in fin:
        if lineがaから始まる:
            char_countsに0をappend
        else:
            lineの末尾の改行文字を除去する
            char_countsの最終要素に文字数を足す

Comments

[22Go]

文字列はできるだけreadlineでypython読み込みたいと思ってます。
リストにするにはstartswithで分けたらいいでしょうか？

[LouiS0616]

readline
その方がメモリを圧迫しませんが、状態を記憶させるための変数が必要になります。

startswith
正規表現を使うと楽で、なおかつ応用が効きます。回答に追記しました。

[LouiS0616]

ごっそり読み込まずとも、わりあいシンプルに書けそうなことに気付きました。
追記しましたので、ご参照ください。

[22Go]

ありがとうございます。
コードもできたら教えていただきたいです。
お手数おかけして申し訳ありません。

[22Go]

-sampledate-
abcde（１）
11111
11111
11

abcde（２）
111

-python-
with open("sampledate.txt") as fin:
char_counts = []
for line in fin:
if line.startswith("a"):
char_counts.append([0])
else:
line.strip()
print(len(char_counts))

-結果-
1
1
1
1
2

でした。どこが間違っているでしょうか。。。

[LouiS0616]

『char_countsの最終要素に文字数を足す』べきところで、『文字数を出力している』だけだからです。
また、char_counts.append([0]) ではなく char_counts.append(0) の方が扱いやすいかと。

[22Go]

すみません、sumやcount使ってみましたが、どうしても導き出せません。。。

[LouiS0616]

難しく考えすぎているような気がします。
char_counts[-1] += len(line) で充分です。

ただ、空白文字(スペースや改行類)のように目に見えない文字が含まれている場合もあることにご注意ください。
rstripについて調べれば、除去の仕方は分かります。

[22Go]

print(char_counts[-1] += len(line))

=にinvalid syntaxがついてしまい、=を消して＋のみで
出力したところ

６
６
３
１
４

と出力されました。。。
rstripについては理解できました。

[LouiS0616]

pythonでの代入は式ではなく文なので、関数の引数に渡すことはできません。
せっかくリストを作っているんですから、その中身はループ脱出後に覗けば良いです。

[22Go]

ありがとうございます。
もう少し工夫してみようかと思います。