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.

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