將Scala映射轉換為Python for Spark

Question

以隨機森林示例為例：

這是一些背景

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

1. 我不確定Map（10-> 4，4，11-> 40）是什么意思？
2. python或pyspark的等效項是什么？

Answer 1

這本質上等同於Python字典

所以

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

成為

categoricalFeaturesInfo={10: 4, 11: 40}