sekvence

Vygenerujte posloupnost celých čísel od počáteční hodnoty ke koncové hodnotě s přírůstkem podle zadaného kroku. Pokud není krok nastaven, funkce zvýší o 1, pokud je počáteční hodnota menší nebo rovna zastavení, jinak se sníží o 1.

Syntaxe

from pyspark.sql import functions as sf

sf.sequence(start, stop, step=None)

Parametry

Parameter Typ Description
start pyspark.sql.Column nebo str Počáteční hodnota (včetně) sekvence.
stop pyspark.sql.Column nebo str Poslední hodnota (včetně) sekvence.
step pyspark.sql.Column nebo str, volitelné Hodnota, která se má přidat k aktuálnímu prvku, aby získal další prvek v sekvenci. Výchozí hodnota je 1, pokud je počáteční hodnota menší nebo rovna zastavení, jinak -1.

Návraty

pyspark.sql.Column: Nový sloupec, který obsahuje matici sekvenčních hodnot.

Examples

Příklad 1: Generování sekvence s výchozím krokem

import pyspark.sql.functions as sf
df = spark.createDataFrame([(-2, 2)], ['start', 'stop'])
df.select(sf.sequence(df.start, df.stop)).show()
+---------------------+
|sequence(start, stop)|
+---------------------+
|    [-2, -1, 0, 1, 2]|
+---------------------+

Příklad 2: Generování sekvence s vlastním krokem

import pyspark.sql.functions as sf
df = spark.createDataFrame([(4, -4, -2)], ['start', 'stop', 'step'])
df.select(sf.sequence(df.start, df.stop, df.step)).show()
+---------------------------+
|sequence(start, stop, step)|
+---------------------------+
|          [4, 2, 0, -2, -4]|
+---------------------------+

Příklad 3: Generování sekvence se záporným krokem

import pyspark.sql.functions as sf
df = spark.createDataFrame([(5, 1, -1)], ['start', 'stop', 'step'])
df.select(sf.sequence(df.start, df.stop, df.step)).show()
+---------------------------+
|sequence(start, stop, step)|
+---------------------------+
|            [5, 4, 3, 2, 1]|
+---------------------------+