每增加一行，按索引增加一行

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()

我希望索引行不断增加