как определить первую запись в кадре данных и выполнить кумулятивное сложение

Question

У меня есть датафрейм, и я должен реализовать приведенный ниже лог c в pyspark

if first.item then
        do;
            cum_count = amt1;
        end;
    else
        do;
            cum_count = cum_count + amt1;
        end;