Проблема: < /h2>
Я хочу распределить значение, которое может быть положительным или отрицательным от одной строки в несколько строк, где каждая строка может содержать только определенное количество, если значение для Распространение положительно, что он заполняет строки, если это отрицательно, это «незаконно». Можно заполнить/не пускать в строку, а порядок наполнения имеет значение, первая пустая или не полностью полная строка - это первое, что вы заполняете, последний заполненный или частично заполненный - это первое, что вы неприятно или частично непристойно.
ожидаемое поведение: < /h2>
Вот пример того, что я ожидаю: сначала появится таблица платежей, где будет одна запись на ACC < /код > и его соответствующий val для распространения.
| acc | val |
|-----|-------|
| 1 | -100 |
| 2 | -123 |
| 3 | -75 |
| 4 | -300 |
| 5 | -77 |
| 6 | -500 |
| 7 | 111 |
| 8 | 123 |
| 9 | 300 |
| 10 | 75 |
< /code>
Тогда вот таблица счетов, чтобы заполнить или непристойен: < /p>
| acc | val | sum |
|-----|------|------|
| 1 | 100 | 100 |
| 1 | 100 | 50 |
| 1 | 100 | 0 |
| 2 | 123 | 123 |
| 2 | 150 | 0 |
| 2 | 123 | 0 |
| 3 | 70 | 70 |
| 3 | 70 | 5 |
| 4 | 150 | 150 |
| 4 | 100 | 100 |
| 4 | 150 | 150 |
| 5 | 77 | 78 |
| 6 | 500 | 500 |
| 6 | 500 | 500 |
| 6 | 500 | 500 |
| 6 | 500 | 500 |
| 7 | 100 | 10 |
| 7 | 100 | 0 |
| 7 | 100 | 0 |
| 8 | 123 | 0 |
| 8 | 123 | 0 |
| 8 | 123 | 0 |
| 9 | 100 | 0 |
| 9 | 150 | 0 |
| 9 | 200 | 0 |
| 10 | 75 | 75 |
| 10 | 75 | 0 |
< /code>
, пока это мой ожидаемый вывод: < /p>
| acc | val | sum |
|-----|------|------|
| 1 | 100 | 50 |
| 1 | 100 | 0 |
| 1 | 100 | 0 |
| 2 | 123 | 0 |
| 2 | 150 | 0 |
| 2 | 123 | 0 |
| 3 | 70 | 0 |
| 3 | 70 | 0 |
| 4 | 150 | 100 |
| 4 | 100 | 0 |
| 4 | 150 | 0 |
| 5 | 77 | 1 |
| 6 | 500 | 500 |
| 6 | 500 | 500 |
| 6 | 500 | 500 |
| 6 | 500 | 0 |
| 7 | 100 | 100 |
| 7 | 100 | 22 |
| 7 | 100 | 0 |
| 8 | 123 | 123 |
| 8 | 123 | 0 |
| 8 | 123 | 0 |
| 9 | 100 | 100 |
| 9 | 150 | 150 |
| 9 | 200 | 50 |
| 10 | 75 | 75 |
| 10 | 75 | 75 |
Что я попробовал:
Это мой текущий подход, но я изо всех сил пытался заставить его работать, столбец res - это где я храню ожидаемую сумму
import polars as pl
pl.Config.set_tbl_rows(100)
pl.Config.set_tbl_cols(20)
left = pl.DataFrame({
"acc": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
"val": [-100, -123, -75, -300, -77, -500, 111, 123, 300, 75]
})
right = pl.DataFrame({
"acc": [1, 1, 1, 2, 2, 2, 3, 3, 4, 4, 4, 5, 6, 6, 6, 6, 7, 7, 7, 8, 8, 8, 9, 9, 9, 10, 10],
"val": [100, 100, 100, 123, 150, 123, 70, 70, 150, 100, 150, 77, 500, 500, 500, 500, 100, 100, 100, 123, 123, 123, 100, 150, 200, 10, 10],
"sum": [100, 50, 0, 123, 0, 0, 70, 5, 150, 100, 150, 78, 500, 500, 500, 500, 10, 0, 0, 123, 0, 0, 0, 0, 0, 0, 0]
})
df = right.join(left.rename({'val': 'bill'}), on='acc')
bill = pl.col('bill')
available = pl.col('val').cum_sum().over('acc')
covered = pl.col('sum').cum_sum(reverse=True).over('acc')
val = pl.when(bill < 0).then(covered + bill).otherwise(available - bill)
results = df.with_columns(bill=bill, available=available, covered=covered, res=val)
print(results)
Подробнее здесь: https://stackoverflow.com/questions/793 ... -condition
