Разделить фрейм данных в соответствии с подсписками по значению обрезки ⇐ Python

[Anonymous]

Я хочу разделить фрейм данных в соответствии с подсписками, полученными в результате разделения списка на части, где единственное значение выше порогового значения является первым.
например. Ограничение = 3
[4,2,3,5,2,1,6,7] => [4,2,3], [5,2,1], [6], [7]
Мне все еще нужно отслеживать другие поля в кадре данных.
Я должен получить заданный результат из этого df< /p>
data = {
"uid": ["Alice", "Bob", "Charlie"],
"time_deltas": [
[4,2, 3],
[1,1, 4, 8, 3],
[1,1, 7, 3, 2],
],
"other_field": [["x", "y", "z"], ["x", "y", "z", "x", "y"], ["x", "y", "z", "x", "y"]]
}

df = pl.DataFrame(data)
cutoff = 3

# Split the time_delta column into lists where the maximum time_delta (excluding the first value) is greater than the cutoff. Ensure that the other_field column is also split accordingly.

# Expected Output
# +--------+----------------------+----------------------+
# | uid | time_deltas | other_field |
# | --- | --- | --- |
# | str | list[duration[ms]] | list[str] |
# +--------+----------------------+----------------------+
# | Alice | [4, 2, 3] | ["x", "y", "z"] |
# | Bob | [1, 1] | ["x", "y"] |
# | Bob | [4] | ["z"] |
# | Bob | [8, 3] | ["x", "y"] |
# | Charlie| [1, 1] | ["x", "y"] |
# | Charlie| [7,3,2] | ["z", "x", "y"] |


Подробнее здесь: https://stackoverflow.com/questions/791 ... toff-value