ParticionarPor(Janela)

Cria um WindowSpec com o particionamento definido.

Sintaxe

Window.partitionBy(*cols)

Parâmetros

Parâmetro	Tipo	Descrição
cols	str, coluna ou lista	Nomes de colunas ou expressões.

Parâmetro	Tipo	Descrição
cols	str, coluna ou lista	Nomes de colunas ou expressões.

Devoluções

WindowSpec

Exemplos

Python

from pyspark.sql import Window, functions as sf

df = spark.createDataFrame(
    [(1, "a"), (1, "a"), (2, "a"), (1, "b"), (2, "b"), (3, "b")], ["id", "category"])

# Show row number ordered by id in each category partition.
window = Window.partitionBy("category").orderBy("id")
df.withColumn("row_number", sf.row_number().over(window)).show()
# +---+--------+----------+
# | id|category|row_number|
# +---+--------+----------+
# |  1|       a|         1|
# |  1|       a|         2|
# |  2|       a|         3|
# |  1|       b|         1|
# |  2|       b|         2|
# |  3|       b|         3|
# +---+--------+----------+