推荐学习书目
› Learn Python the Hard Way
Python Sites
› PyPI - Python Package Index
› http://diveintopython.org/toc/index.html
› Pocoo
值得关注的项目
› PyPy
› Celery
› Jinja2
› Read the Docs
› gevent
› pyenv
› virtualenv
› Stackless Python
› Beautiful Soup
› 结巴中文分词
› Green Unicorn
› Sentry
› Shovel
› Pyflakes
› pytest
Python 编程
› pep8 Checker
Styles
› PEP 8
› Google Python Style Guide
› Code Style from The Hitchhiker's Guide
这是一个创建于 1492 天前的主题，其中的信息可能已经有所发展或是发生改变。
比如［-1，-1，-1，1，1，1，1，-1，-1 ］。
能变成 -1，-2，-3，1，2，3，4，-1，-2
最好是高效率的
能变成 -1，-2，-3，1，2，3，4，-1，-2
最好是高效率的
 |
1
necomancer 2020-10-30 16:40:46 +08:00
a = np.array([-1,-1,-1,1,1,1,1,-1,-1])
bounds = np.r_[0, np.argwhere(np.diff(a>0, prepend=0)).ravel(), a.shape[0]] np.concatenate([np.cumsum(a[i:j]) for i, j in zip(bounds[:-1], bounds[1:])]) |
|
2
necomancer 2020-10-30 16:53:32 +08:00
可能快一点，尤其是 1， -1 交错很厉害的时候
import pandas as pd b = pd.Series(np.insert(a, np.argwhere(np.diff(a>0, prepend=0)).ravel(), 0)) cum = b.cumsum() cum = (cum -cum.where(b==0).ffill().fillna(0)).values cum = cum[cum!=0] |
 |
3
admirez OP @necomancer 非常感谢，我试试看
|
Select Language