python - convert loc expression from pandas to Pyspark?

Question

How can I convert this expression from pandas to Pyspark Dataframe?

The target is to assign the column date_stamp the value cur

#the data frame is:
tag,     2020-06-25
-------------------
3FMTK1RM    0
678jhgt     18

#######################
vin='3FMTK1RM'# is the first element of tag
cur= 5
date_stamp='2020-06-25'
df.loc[str(date_stamp),vin] = cur

question from:https://stackoverflow.com/questions/66061625/convert-loc-expression-from-pandas-to-pyspark

Answer 1

You can use when:

import pyspark.sql.functions as F

df2 = df.withColumn(
    '2020-06-25', 
    F.when(F.col('tag') == 'vin', cur).otherwise(F.col('2020-06-25'))
)