如何从PySpark的PCAModel对象中获取主成分数量？

嘿，这个问题我之前也踩过坑！你用pca_model.k或者getParam('k')拿到的其实是参数的定义对象，不是实际的数值，所以才会看到那段Param的描述文本。下面给你几个靠谱的解决方法：

方法1：用getOrDefault('k')直接获取参数值 #

这是最直接的方式，getOrDefault方法会返回参数的实际设置值（如果没手动设置就用参数的默认值）：

# 输出你设置的k值，这里就是5
print(pca_model.getOrDefault('k'))

方法2：通过extractParamMap()获取参数键值对 #

如果你需要查看多个参数的取值，这个方法可以拿到所有参数的键值映射，再从中提取k的数值：

param_map = pca_model.extractParamMap()
# 从参数映射中取出k对应的数值
print(param_map[pca_model.k])

方法3：通过主成分矩阵的形状验证真实数量 #

如果你想确认模型实际生成的主成分数量（比如当输入特征数小于你设置的k时，模型会自动用特征数作为主成分数），可以查看pc属性的形状——主成分矩阵的行数就是实际的主成分数量：

# 输出模型实际生成的主成分数量
print(pca_model.pc.shape[0])

比如如果你的输入特征只有3个，但设置了k=5，模型实际只会生成3个主成分，这时候用这个方法能得到最准确的结果。

内容的提问来源于stack exchange，提问作者Clock Slave