Question

dataframe 增加一列（这一行1连续出现的次数）

在dataframe列表的最后增加一列（这一行1连续出现的次数）

Answer 1

def checknum(l,n=1):
#计算列表中连续=n的数目，返回最大连续数
res=[]
count=0
for i in l:
if i == n:
count+=1
else:
res.append(count)
count=0
res.append(count)
return max(res)

d=[
[1, 0, 0,1],
[1, 1, 1,1],
[1, 1, 0,0],
[1, 1, 1,1],
[0, 0, 0,1]
]

print [i+[checknum(i)] for i in d] #计算每行连续出现1的次数并加到行最后

输出结果>>>
[[1, 0, 0, 1, 1], [1, 1, 1, 1, 4], [1, 1, 0, 0, 2], [1, 1, 1, 1, 4], [0, 0, 0, 1, 1]]