Pular para o conteúdo principal

soma (DadosAgrupados)

Calcule a soma de cada coluna numérica para cada grupo.

Sintaxe

sum(*cols)

Parâmetros

Parâmetro

Tipo

Descrição

cols

str

Nomes das colunas. Colunas não numéricas são ignoradas.

Parâmetro

Tipo

Descrição

cols

str

Nomes das colunas. Colunas não numéricas são ignoradas.

Devoluções

DataFrame

Exemplos

Python
df = spark.createDataFrame([
(2, "Alice", 80), (3, "Alice", 100),
(5, "Bob", 120), (10, "Bob", 140)], ["age", "name", "height"])

# Group-by name, and calculate the sum of the age in each group.
df.groupBy("name").sum("age").sort("name").show()
# +-----+--------+
# | name|sum(age)|
# +-----+--------+
# |Alice| 5|
# | Bob| 15|
# +-----+--------+

# Calculate the sum of the age and height in all data.
df.groupBy().sum("age", "height").show()
# +--------+-----------+
# |sum(age)|sum(height)|
# +--------+-----------+
# | 20| 440|
# +--------+-----------+