size

Função de coleção: retorna o comprimento da matriz ou mapa armazenado na coluna. Dá suporte ao Spark Connect.

Para a função SQL do Databricks correspondente, consulte a função size.

Sintaxe

from pyspark.databricks.sql import functions as dbf

dbf.size(col=<col>)

Parâmetros

Parâmetro	Tipo	Description
col	pyspark.sql.Column ou str	Nome da coluna ou expressão.

Devoluções

pyspark.sql.Column: comprimento da matriz/mapa.

Exemplos

from pyspark.databricks.sql import functions as dbf
df = spark.createDataFrame([([1, 2, 3],),([1],),([],)], ['data'])
df.select(dbf.size(df.data)).collect()

[Row(size(data)=3), Row(size(data)=1), Row(size(data)=0)]