Add integer in repeating N rows based on a condition for a PySpark DataFrame

Question

I am provided with the following DataFrame with panel data containing Dates, Stocks and some feature for each stock.

I would like to create a new PySpark data