• 功能介绍
  • 参数说明
  • 脚本示例
    • 脚本
    • 结果

    功能介绍

    相关系数算法用于计算一个矩阵中每一列之间的相关系数，范围在[-1,1]之间。计算的时候，count数按两列间同时非空的元素个数计算，两两列之间可能不同。

    参数说明

    名称 中文名称 描述 类型 是否必须？ 默认值
    selectedCols 选中的列名数组 计算列对应的列名列表 String[] null
    method 方法 方法：包含”pearson”和”spearman”两种，默认值是pearson。 String “pearson”

    脚本示例

    脚本

    1. data = np.array([
    2.          [0.0,0.0,0.0],
    3.          [0.1,0.2,0.1],
    4.          [0.2,0.2,0.8],
    5.          [9.0,9.5,9.7],
    6.          [9.1,9.1,9.6],
    7.          [9.2,9.3,9.9]])
    9. df = pd.DataFrame({"x1": data[:, 0], "x2": data[:, 1], "x3": data[:, 2]})
    10. source = dataframeToOperator(df, schemaStr='x1 double, x2 double, x3 double', op_type='batch')
    13. corr = CorrelationBatchOp()\
    14.             .setSelectedCols(["x1","x2","x3"])
    16. correlation = source.link(corr).collectCorrelationResult()
    17. print(correlation.getCorrelation())

    结果

    1. [[1.0, 0.9994126290576466, 0.9990251498141454], [0.9994126290576466, 1.0, 0.9985989260453877], [0.9990251498141454, 0.9985989260453877, 1.0]]