Pyspark: substituindo valor em uma coluna pesquisando um dicionário

Eu sou um novato no PySpark.

Eu tenho uma faíscaDataFrame df que possui uma coluna 'device_type'.

Desejo substituir todos os valores contidos em "Tablet" ou "Telefone" por "Telefone" e substituir "PC" por "Área de trabalho".

No Python, posso fazer o seguinte,

deviceDict = {'Tablet':'Mobile','Phone':'Mobile','PC':'Desktop'}
df['device_type'] = df['device_type'].replace(deviceDict,inplace=False)

Como posso conseguir isso usando o PySpark? Obrigado!

questionAnswers(3)

yourAnswerToTheQuestion