Pythonで、リスト内の特定値が要素の頭から何回連続で出現するかを数えたいです

Question

###前提・実現したいこと
pythonでCSVファイルの各行をリストとして取り出し、
リスト内の特定値が要素の頭から何回連続で出現するかを数えたいです

例えば特定値が0として、csvが

data1,data2,data3,data4
0,0,0,0
0,1,3,0
1,0,4,1

とするならば、

data1,data2,data3,data4,count
0,0,0,0,4
0,1,3,0,1
1,0,4,1,0

といった感じです。

###試したこと
for文とif文を使ってコードを書こうとしたのですが、条件が入れ子状になってしまい
考えがまとまらず実装できませんでした。。。

###補足情報(言語/FW/ツール等のバージョンなど)
pandasは多少使えますので、python,numpy,pandasあたりを用いる内容だと有難いです。

Answer 1

行の頭から0 が何回続くかをカウントするってことですかね？
pandasの場合こんな感じでしょうか

数値にマイナスがある場合は使えない手ですが・・。

import pandas as pd

df = pd.DataFrame([[0,0,0,0],[0,1,3,0],[1,0,4,1]])
df['count'] = (df.cumsum(axis=1) == 0).sum(axis=1)
print(df)
#=>   0  1  2  3  count
#  0  0  0  0  0      4
#  1  0  1  3  0      1
#  2  1  0  4  1      0

Comments

[magichan]

あっ、データにマイナスがあっても、abs() 挟めば大丈夫かも。

[Phillip]

分かりやすい回答、ありがとうございました！
cumsumを使ったことが無かったので、とても参考になりました。

Answer 2

Pandasなし版も。

from csv import reader
from functools import partial
from itertools import takewhile
from operator import eq


csv = reader('''\
data1,data2,data3,data4
0,0,0,0
0,1,3,0
1,0,4,1
'''.splitlines())
eq_0 = partial(eq, '0')

print(*next(csv), 'count', sep=',')
for row in csv:
    print(*row, len([*takewhile(eq_0, row)]), sep=',')

Answer 3

文字列に変換して正規表現で頑張ってみます。

import re
import pandas as pd
df = pd.DataFrame([[0,0,0,0],[0,1,3,0],[1,0,4,1]])
df = df.astype('str')
col = df.columns
b = df[col].apply(lambda x: ''.join(x), axis=1)
key = '0'
df['count'] = b.apply(lambda x: len(re.match('^[{0}]*'.format(key), x).group(0))//len(key)).values