Создайте столбец, который присваивает «1» первым n наблюдениям, «2» следующим n наблюдениям и т. Д. В pyspark.sql.dataframe.DataFrame

Question

У меня есть pyspark.sql.dataframe, который выглядит так n наблюдений, от '2' до следующих n наблюдений и т. Д., Так что я получаю такой кадр данных ( n = 2 в этом примере)

ID   Score   Segment
5    0.7     1
4    0.5     1
1    0.4     2
3    0.35    2
2    0.3     3
6    0.2     3

Я смотрю на pyspark.sql.Window, но не могу найти хороший способ реализовать его.Спасибо!