Communities

Writing
Writing
Codidact Meta
Codidact Meta
The Great Outdoors
The Great Outdoors
Photography & Video
Photography & Video
Scientific Speculation
Scientific Speculation
Cooking
Cooking
Electrical Engineering
Electrical Engineering
Judaism
Judaism
Languages & Linguistics
Languages & Linguistics
Software Development
Software Development
Mathematics
Mathematics
Christianity
Christianity
Code Golf
Code Golf
Music
Music
Physics
Physics
Linux Systems
Linux Systems
Power Users
Power Users
Tabletop RPGs
Tabletop RPGs
Community Proposals
Community Proposals
tag:snake search within a tag
answers:0 unanswered questions
user:xxxx search by author id
score:0.5 posts with 0.5+ score
"snake oil" exact phrase
votes:4 posts with 4+ votes
created:<1w created < 1 week ago
post_type:xxxx type of post
Search help
Notifications
Mark all as read See all your notifications »
Q&A

Welcome to Software Development on Codidact!

Will you help us build our independent community of developers helping developers? We're small and trying to grow. We welcome questions about all aspects of software development, from design to code to QA and more. Got questions? Got answers? Got code you'd like someone to review? Please join us.

Post History

50%
+0 −0
Q&A How to get conditional running cumulative sum based on current row and previous rows?

Use a collect_list collecting the values from all preceding rows up to the current row into a struct Then filter on that struct based on its value and the current row's threshold Use aggregate ...

posted 6mo ago by congusbongus‭

Answer
#1: Initial revision by user avatar congusbongus‭ · 2024-07-02T01:30:01Z (6 months ago)
- Use a [`collect_list`](https://www.sparkreference.com/reference/collect_list/) collecting the values from all preceding rows up to the current row into a struct
- Then [`filter`](https://spark.apache.org/docs/latest/api/sql/index.html#filter) on that struct based on its value and the current row's threshold
- Use [`aggregate`](https://spark.apache.org/docs/latest/api/sql/index.html#aggregate) to calculate the result based on adding the struct's `acc` field

Note that doing so may reorder the output so I added an order column.

```python
import pyspark.sql.functions as F

df = spark.createDataFrame([(1, 3, 1, 1), (2, 1, 2, 2), (3, 2, 3, 2)], ["order", "acc", "value", "threshold"])
display(
  df
  .withColumn("output", F.expr("""
  aggregate(
    filter(
      collect_list(struct(acc, value)) OVER (ORDER BY order ROWS BETWEEN UNBOUNDED PRECEDING AND CURRENT ROW), 
      s -> s.value >= threshold
    ),
    0L, 
    (output, s) -> output + s.acc
  )
  """))
  .orderBy("order")
)
```

| order | acc | value | threshold | output |
|-------|-----|-------|-----------|--------|
| 1     | 3   | 1     | 1         | 3      |
| 2     | 1   | 2     | 2         | 1      |
| 3     | 2   | 3     | 2         | 3      |