pyspark.sql.functions.corr

pyspark.sql.functions.corr(col1, col2)

Returns a new Column for the Pearson Correlation Coefficient for col1 and col2.

New in version 1.6.0.

Examples

>>> a = range(20)
>>> b = [2 * x for x in range(20)]
>>> df = spark.createDataFrame(zip(a, b), ["a", "b"])
>>> df.agg(corr("a", "b").alias('c')).collect()
[Row(c=1.0)]

pyspark.sql.functions.conv pyspark.sql.functions.cos