fatia

Retorna uma nova coluna de matriz segmentando a coluna de matriz de entrada de um índice inicial até um comprimento específico. Os índices começam em 1 e podem ser negativos para indexar a partir do final da matriz. O comprimento especifica o número de elementos na matriz resultante.

Sintaxe

from pyspark.sql import functions as sf

sf.slice(x, start, length)

Parâmetros

Parâmetro	Tipo	Description
x	pyspark.sql.Column ou str	Coluna de matriz de entrada ou nome de coluna a ser fatiado.
start	pyspark.sql.Column, str ou int	O índice inicial da operação de fatia. Se negativo, inicia o índice a partir do final da matriz.
length	pyspark.sql.Column, str ou int	O comprimento da fatia, representando o número de elementos na matriz resultante.

Devoluções

pyspark.sql.Column: um novo objeto Column do tipo Array, em que cada valor é uma fatia da lista correspondente da coluna de entrada.

Exemplos

Exemplo 1: Uso básico da função de fatia.

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

+--------------+
|slice(x, 2, 2)|
+--------------+
|        [2, 3]|
|           [5]|
+--------------+

Exemplo 2: fatiamento com índice inicial negativo.

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

+---------------+
|slice(x, -1, 1)|
+---------------+
|            [3]|
|            [5]|
+---------------+

Exemplo 3: função Slice com entradas de coluna para início e comprimento.

from pyspark.sql import functions as sf
df = spark.createDataFrame([([1, 2, 3], 2, 2), ([4, 5], 1, 3)], ['x', 'start', 'length'])
df.select(sf.slice(df.x, df.start, df.length)).show()

+-----------------------+
|slice(x, start, length)|
+-----------------------+
|                 [2, 3]|
|                 [4, 5]|
+-----------------------+