count

Devuelve el número de elementos de un grupo.

Syntax

from pyspark.sql import functions as sf

sf.count(col)

Parámetros

Parámetro	Tipo	Description
col	pyspark.sql.Column o nombre de columna	Columna de destino en la que se va a calcular.

Devoluciones

pyspark.sql.Column: columna para los resultados calculados.

Examples

Ejemplo 1: Recuento de todas las filas de un dataframe

from pyspark.sql import functions as sf
df = spark.createDataFrame([(None,), ("a",), ("b",), ("c",)], schema=["alphabets"])
df.select(sf.count(sf.expr("*"))).show()

+--------+
|count(1)|
+--------+
|       4|
+--------+

Ejemplo 2: Recuento de valores no NULL en una columna específica

from pyspark.sql import functions as sf
df.select(sf.count(df.alphabets)).show()

+----------------+
|count(alphabets)|
+----------------+
|               3|
+----------------+

Ejemplo 3: Recuento de todas las filas de un dataframe con varias columnas

from pyspark.sql import functions as sf
df = spark.createDataFrame(
    [(1, "apple"), (2, "banana"), (3, None)], schema=["id", "fruit"])
df.select(sf.count(sf.expr("*"))).show()

+--------+
|count(1)|
+--------+
|       3|
+--------+

Ejemplo 4: Recuento de valores no NULL en varias columnas

from pyspark.sql import functions as sf
df.select(sf.count(df.id), sf.count(df.fruit)).show()

+---------+------------+
|count(id)|count(fruit)|
+---------+------------+
|        3|           2|
+---------+------------+