array_size

Gibt die Gesamtanzahl der Elemente im Array zurück. Die Funktion gibt NULL für null-Eingaben zurück.

Syntax

from pyspark.sql import functions as sf

sf.array_size(col)

Die Parameter

Parameter	Typ	Description
col	pyspark.sql.Column oder str	Der Name der Spalte oder eines Ausdrucks, der das Array darstellt.

Rückkehr

pyspark.sql.Column: Eine neue Spalte, die die Größe der einzelnen Arrays enthält.

Examples

Beispiel 1: Einfache Verwendung mit ganzzahligem Array

from pyspark.sql import functions as sf
df = spark.createDataFrame([([2, 1, 3],), (None,)], ['data'])
df.select(sf.array_size(df.data)).show()

+----------------+
|array_size(data)|
+----------------+
|               3|
|            NULL|
+----------------+

Beispiel 2: Verwendung mit Zeichenfolgenarray

from pyspark.sql import functions as sf
df = spark.createDataFrame([(['apple', 'banana', 'cherry'],)], ['data'])
df.select(sf.array_size(df.data)).show()

+----------------+
|array_size(data)|
+----------------+
|               3|
+----------------+

Beispiel 3: Verwendung mit gemischten Typenarray

from pyspark.sql import functions as sf
df = spark.createDataFrame([(['apple', 1, 'cherry'],)], ['data'])
df.select(sf.array_size(df.data)).show()

+----------------+
|array_size(data)|
+----------------+
|               3|
+----------------+

Beispiel 4: Verwendung mit Arrays von Arrays

from pyspark.sql import functions as sf
df = spark.createDataFrame([([[2, 1], [3, 4]],)], ['data'])
df.select(sf.array_size(df.data)).show()

+----------------+
|array_size(data)|
+----------------+
|               2|
+----------------+

Beispiel 5: Verwendung mit leerem Array

from pyspark.sql import functions as sf
from pyspark.sql.types import ArrayType, IntegerType, StructType, StructField
schema = StructType([
  StructField("data", ArrayType(IntegerType()), True)
])
df = spark.createDataFrame([([],)], schema=schema)
df.select(sf.array_size(df.data)).show()

+----------------+
|array_size(data)|
+----------------+
|               0|
+----------------+