我正在尝试将python代码转换为pyspark，但由于“Row”对象不支持项分配而出现错误

Question

我正在尝试在pyspark中转换python代码。请告诉我pyspark版本的代码有什么问题：-

原始python版本：-

for i in range(0,km_data.count()):

  if i==0:

     km_data['risk'].iloc[i]=not_lapsed+lapsed
      

  else:

    km_data['risk'].iloc[i]=km_data['risk'].iloc[i-1]-(km_data['lapsed'].iloc[i-1])-(km_data['censored'].iloc[i])

使用的Pyspark版本：-

for i in range(0,km_data.count()):

  if i==0:

    km_data.collect()[i]['risk']=not_lapsed+lapsed

  else:

    km_data.collect()[i]['risk']=km_data.collect()[i-1]['risk']-(km_data.collect()[i-1]['lapsed'])-(km_data.collect()[i-1]['cencosred'])

基本上，我正在pyspark中寻找等效的iloc，它可以帮助我获得结果。请忽略缩进问题，因为我使用mobile键入了此代码。

Answer 1

Collect（）可以在这里使用，我可以看到你也做了同样的事情-

X = df.collect()[0]['age']  
or 
X = df.collect()[0][1]  #row 0 col 1 

除了你在找什么，还有别的吗？