Convert SAS program to python $($Without Pandas$)$ :

DATA test;

INPUT prenom :$$9.$;

CARDS;

Mathilde

Fr$$d$$ric

Subira

Fr$$d$$ric

Subira

Subira

;

PROC SORT DATA$=$test;

BY prenom;

RUN;

DATA test$2$;

SET TEST;

BY prenom;

x$1=$first.prenom;

x$2=$last.prenom;

RUN;

Results :

Obs. prenom x$1$ x$2$

$1$ Fr$$d$$ric $10$

$2$ Fr$$d$$ric $01$

$3$ Mathilde $11$

$4$ Subira $10$

$5$ Subira $00$

$6$ Subira $01$

This is what I have already been able to do but it is not the desired result :

df$\_$test$=$ spark.createDataFrame$(["$Mathilde$","$Fr$$d$$ric$",$"Subira","Fr$$d$$ric$",$"Subira","Subira"$],$ "string"$).$toDF$("$pr$$nom$")$

from pyspark.sql import functions as F$,$ Window as W

w $=$ W$.$partitionBy$(\text{'}$pr$$nom$\text{'}).$orderBy$("$pr$$nom$")$

df $=$ df$\_$test.withColumn$(\text{'}$X$\text{'},$ F$.$row$\_$number$().$over$($w$))$