matriz_repetir

Cria uma matriz contendo uma coluna que se repete uma contagem de vezes.

Sintaxe

Python

from pyspark.sql import functions as sf

sf.array_repeat(col, count)

Parâmetros

Parâmetro	Tipo	Descrição
col	pyspark.sql.Column ou str	O nome da coluna ou uma expressão que representa o elemento a ser repetido.
count	pyspark.sql.Column, str ou int	O nome da coluna, uma expressão ou um número inteiro que representa a quantidade de vezes que o elemento deve se repetir.

Devoluções

pyspark.sql.ColumnUma nova coluna que contém uma matriz de elementos repetidos.

Exemplos

Exemplo 1 : Utilização com strings

Python

from pyspark.sql import functions as sf
df = spark.createDataFrame([('ab',)], ['data'])
df.select(sf.array_repeat(df.data, 3)).show()

Output

+---------------------+
|array_repeat(data, 3)|
+---------------------+
|         [ab, ab, ab]|
+---------------------+

Exemplo 2 : Uso com números inteiros

Python

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

Output

+---------------------+
|array_repeat(data, 2)|
+---------------------+
|               [3, 3]|
+---------------------+

Exemplo 3 : Utilização com matriz

Python

from pyspark.sql import functions as sf
df = spark.createDataFrame([(['apple', 'banana'],)], ['data'])
df.select(sf.array_repeat(df.data, 2)).show(truncate=False)

Output

+----------------------------------+
|array_repeat(data, 2)             |
+----------------------------------+
|[[apple, banana], [apple, banana]]|
+----------------------------------+

Exemplo 4 : Uso com valor nulo

Python

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

Output

+---------------------+
|array_repeat(data, 3)|
+---------------------+
|   [NULL, NULL, NULL]|
+---------------------+