pandas - 系列中的元素的索引，其中累积的一些等于或大于 10

Question

我有这个 Panda 整数系列，例如s = pd.Series([1, 2, 3, 1, 5, 10]). 我需要找到从累积和等于或大于某个数字开始的元素的索引，例如10. 所以，对于这个系列，因为print(s.cumsum())给了我们

0     1
1     3
2     6
3     7
4    12
5    22

我需要得到4，因为这里的累积总和大于10。

找到该索引的最佳 Python 方法是什么？

标签： pandasindexingcumsum

Answer 1

只需在idxmax之后添加条件cumsum

(s.cumsum()>10).idxmax()