Bagikan melalui


sequence

Hasilkan urutan bilangan bulat dari awal hingga akhir, tahapan demi langkah. Jika langkah tidak diatur, fungsi bertambah 1 jika mulai kurang dari atau sama dengan berhenti, jika tidak, maka akan mengalami penurunan sebesar 1.

Syntax

from pyspark.sql import functions as sf

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

Parameter-parameternya

Pengaturan Tipe Description
start pyspark.sql.Column atau str Nilai awal (inklusif) dari urutan.
stop pyspark.sql.Column atau str Nilai terakhir (inklusif) dari urutan.
step pyspark.sql.Column atau str, opsional Nilai yang akan ditambahkan ke elemen saat ini untuk mendapatkan elemen berikutnya dalam urutan. Defaultnya adalah 1 jika mulai kurang dari atau sama dengan berhenti, jika tidak -1.

Pengembalian Barang

pyspark.sql.Column: Kolom baru yang berisi array nilai urutan.

Examples

Contoh 1: Menghasilkan urutan dengan langkah default

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]|
+---------------------+

Contoh 2: Menghasilkan urutan dengan langkah kustom

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]|
+---------------------------+

Contoh 3: Menghasilkan urutan dengan langkah negatif

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]|
+---------------------------+