Ai
1 Star 0 Fork 0

qianyun6/python_utils

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
pandas_utils.py 574 Bytes
一键复制 编辑 原始数据 按行查看 历史
qianyun6 提交于 2018-04-24 21:27 +08:00 . util 1
import pandas as pd
import numpy as np
def count_consecutive(boolseries):
"""
Count consecutive appearance of True in a pandas Series of bool
Param:
boolseries: pandas.Series of type bool
Return: pandas.Series of type int
Number of consecutive True if corresponding position is True else 0
"""
grp = boolseries.groupby((boolseries != boolseries.shift()).cumsum())
return pd.Series(data=np.where(boolseries, grp.cumcount() + grp.cumcount(ascending=False) + 1, 1),
index=boolseries.index
)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/qianyun6/python_utils.git
git@gitee.com:qianyun6/python_utils.git
qianyun6
python_utils
python_utils
master

搜索帮助