ичок в PySpark.

У меня есть искраDataFrame df который имеет столбец «тип_устройства».

Я хочу заменить каждое значение в «Планшете» или «Телефоне» на «Телефон» и заменить «ПК» на «Рабочий стол».

В Python я могу сделать следующее,

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

Как я могу добиться этого с помощью PySpark? Спасибо!