Pyspark: Reemplazar valor en una columna buscando un diccionario

Soy un novato en PySpark.

Tengo una chispaDataFrame df que tiene una columna 'tipo_dispositivo'.

Quiero reemplazar cada valor que está en "Tableta" o "Teléfono" a "Teléfono", y reemplazar "PC" a "Escritorio".

En Python puedo hacer lo siguiente,

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

¿Cómo puedo lograr esto usando PySpark? ¡Gracias!

Respuestas a la pregunta(3)

Su respuesta a la pregunta