Cumsum reset at NaN

by

Even more pandas-onic way to do it:

v = pd.Series([1., 3., 1., np.nan, 1., 1., 1., 1., np.nan, 1.])
cumsum = v.cumsum().fillna(method='pad')
reset = -cumsum[v.isnull()].diff().fillna(cumsum)
result = v.where(v.notnull(), reset).cumsum()

Contrary to the matlab code, this also works for values different from 1.

More Related Contents:

  1. Replacing values represented by 'UN' with NaN
  2. Strings in a DataFrame, but dtype is object
  3. What does axis in pandas mean?
  4. FutureWarning: elementwise comparison failed; returning scalar, but in the future will perform elementwise comparison
  5. pandas select from Dataframe using startswith
  6. How to map numeric data into categories / bins in Pandas dataframe
  7. What’s the fastest way in Python to calculate cosine similarity given sparse matrix data?
  8. Cumsum as a new column in an existing Pandas data
  9. Restart cumsum and get index if cumsum more than value
  10. How can I map True/False to 1/0 in a Pandas DataFrame?
  11. What does the term “broadcasting” mean in Pandas documentation?
  12. Dropping infinite values from dataframes in pandas?
  13. ValueError: numpy.dtype has the wrong size, try recompiling
  14. Convert timedelta64[ns] column to seconds in Python Pandas DataFrame
  15. ‘DataFrame’ object has no attribute ‘sort’
  16. Pandas pd.Series.isin performance with set versus array
  17. Difference between data type ‘datetime64[ns]’ and ‘
  18. Missing data, insert rows in Pandas and fill with NAN
  19. python pandas flatten a dataframe to a list
  20. Precision lost while using read_csv in pandas
  21. Get week start date (Monday) from a date column in Python (pandas)?
  22. How can I use cumsum within a group in Pandas?
  23. Pandas dataframe – running sum with reset
  24. What is dtype(‘O’), in pandas?
  25. pandas distinction between str and object types
  26. NumPy version of “Exponential weighted moving average”, equivalent to pandas.ewm().mean()
  27. Python Pandas – Missing required dependencies [‘numpy’] 1
  28. Pandas: Difference between largest and smallest value within group
  29. How to use sklearn fit_transform with pandas and return dataframe instead of numpy array?
  30. mean from pandas and numpy differ

Leave a Comment