pyspark.sql.functions.collect_set

pyspark.sql.functions.collect_set(col)[source]

Aggregate function: returns a set of objects with duplicate elements eliminated.

New in version 1.6.0.

Notes

The function is non-deterministic because the order of collected results depends on the order of the rows which may be non-deterministic after a shuffle.

Examples

>>> df2 = spark.createDataFrame([(2,), (5,), (5,)], ('age',))
>>> df2.agg(collect_set('age')).collect()
[Row(collect_set(age)=[5, 2])]