如何在Spark DataFrame中获取分类变量的索引？

Question

我使用python创建了一个Spark DataFrame，我希望获得分类变量的映射。换句话说，我想获得与在python pandas上运行df['name-of-variable'].cat.codes相同的结果。有没有办法实现这个目标？