Add a column based on value of another column in Pyspark

Question

I have a Spark dataframe with a column A and I want to append a column B based on the values in column A. It is established that the v