В кадре данных pyspark я хочу подсчитать отдельные значения на основе окна ⇐ Python

[Гость]

У меня есть фрейм данных, например


< tr>
план
имя
user_id
долларов




FWE
QRS
123
10


FWE
QRS
123
20

FWE
QRS
223
30


RWE
QRS
223
40



введите здесь описание изображения
Я хочу посчитать количество уникальных идентификаторов user_id на основе раздела по плану и имени.
Я ожидаю вывода например



< th>план
имя
user_id
долларов
un_count




FWE
QRS
123
10
2
< /tr>

FWE
QRS
123
20
2


FWE
QRS
223
30
1
< /tr>

RWE
QRS
223
40
1



Я хочу, чтобы этот un_count использовался с функцией withColumn

Подробнее здесь: https://stackoverflow.com/questions/781 ... -on-window