добавить дополнительную строку по возрастанию индекса для каждой строки строки

Question

ratings_test = test_data.map(lambda l: l.split()).map(lambda a : 
    Row(userId=int(a[0]),movieId=int(a[1]),index=i)).cache()

Я хочу, чтобы индексная строка увеличивалась