将Scala映射转换为Python for Spark

Question

作为随机森林的例子：

这是一些背景

val forest = RandomForest.trainClassifier(
  trainData, 7, Map(10 -> 4, 11 -> 40), 20,
  "auto", "entropy", 30, 300)

1. 我不确定地图（10 - ＆gt; 4,11 - ＆gt; 40）是什么意思？
2. 什么是python或pyspark相当于此？

Answer 1

这基本上相当于Python字典

所以

Map(10 -> 4, 11 -> 40)

成为

categoricalFeaturesInfo={10: 4, 11: 40}