pyspark.sql.functions.size

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

Collection function: returns the length of the array or map stored in the column.

New in version 1.5.0.

Parameters:
colColumn or str

name of column or expression

Examples

>>> df = spark.createDataFrame([([1, 2, 3],),([1],),([],)], ['data'])
>>> df.select(size(df.data)).collect()
[Row(size(data)=3), Row(size(data)=1), Row(size(data)=0)]