Compare values with a specific margin of error

Question

Is there a way to compare two values of type double in PySpark, with a specified margin of error? Essential similar to this post, but in PySpark.

Something like:

df=#some dataframe with 2 columns RESULT1 and RESULT2

df=withColumn('compare', when(col('RESULT1')==col('RESULT2') +/- 0.05*col('RESULT2'), lit("match")).otherwise(lit("no match"))

But in a more elegant way?

score 1 · Accepted Answer · answered Feb 08 '21 at 14:10

1

You can use between as the condition:

df2 = df.withColumn(
    'compare',
    when(
        col('RESULT1').between(0.95*col('RESULT2'), 1.05*col('RESULT2')), 
        lit("match")
    ).otherwise(
        lit("no match")
    )
)

answered Feb 08 '21 at 14:10

mck

37,331
13
29
45

score 1 · Answer 2 · answered Feb 08 '21 at 14:36

You can also write as |RESULT1 - RESULT2| <= 0.05 * RESULT2 :

from pyspark.sql import functions as F

df1 = df.withColumn(
    'compare',
    F.when(
        F.abs(F.col('RESULT1') - F.col("RESULT2")) <= 0.05 * F.col("RESULT2"),
        F.lit("match")
    ).otherwise(F.lit("no match"))
)

Compare values with a specific margin of error

2 Answers2