Pular para o conteúdo principal

tamanho

Função de coleção: Retorna o comprimento da matriz ou mapa armazenado na coluna. Compatível com Spark Connect.

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

Sintaxe

Python
from pyspark.databricks.sql import functions as dbf

dbf.size(col=<col>)

Parâmetros

Parâmetro

Tipo

Descrição

col

pyspark.sql.Column ou str

Nome da coluna ou expressão.

Devoluções

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

Exemplos

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