Como adicionar colunas específicas no pandas (com exemplos)

Você pode usar os seguintes métodos para encontrar a soma de um conjunto específico de colunas em um DataFrame do pandas:

Método 1: Encontre a soma de todas as colunas

 #find sum of all columns
df[' sum '] = df. sum (axis= 1 )

Método 2: Encontre a soma de colunas específicas

 #specify the columns to sum
cols = [' col1 ', ' col4 ', ' col5 ']

#find sum of columns specified 
df[' sum '] = df[cols]. sum (axis= 1 )

Os exemplos a seguir mostram como usar cada método na prática com o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

   points assists rebounds
0 18 5 11
1 22 7 8
2 19 7 10
3 14 9 6
4 14 12 6
5 11 9 5
6 20 9 9
7 28 4 12

Exemplo 1: Encontre a soma de todas as colunas

O código a seguir mostra como adicionar os valores das linhas em todas as colunas do DataFrame:

 #define new column that contains sum of all columns
df[' sum_stats '] = df. sum (axis= 1 )

#view updated DataFrame
df

	points assists rebounds sum_stats
0 18 5 11 34
1 22 7 8 37
2 19 7 10 36
3 14 9 6 29
4 14 12 6 32
5 11 9 5 25
6 20 9 9 38
7 28 4 12 44

A coluna sum_stats contém a soma dos valores das linhas em todas as colunas.

Por exemplo, aqui está como os valores foram calculados:

• Soma da linha 0: 18 + 5 + 11 = 34
• Soma da linha 1: 22 + 7 + 8 = 37
• Soma da linha 2: 19 + 7 + 10 = 36

E assim por diante.

Exemplo 2: Encontre a soma de colunas específicas

O código a seguir mostra como adicionar os valores das linhas em todas as colunas do DataFrame:

 #specify the columns to sum
cols = [' points ', ' assists ']

#define new column that contains sum of specific columns
df[' sum_stats '] = df[cols]. sum (axis= 1 )

#view updated DataFrame
df

	points assists rebounds sum_stats
0 18 5 11 23
1 22 7 8 29
2 19 7 10 26
3 14 9 6 23
4 14 12 6 26
5 11 9 5 20
6 20 9 9 29
7 28 4 12 32

A coluna sum_stats contém a soma dos valores das linhas nas colunas “pontos” e “assistências”.

Por exemplo, aqui está como os valores foram calculados:

• Soma da linha 0: 18 + 5 + 11 = 23
• Soma da linha 1: 22 + 7 = 29
• Soma da linha 2: 19 + 7 = 26

E assim por diante.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como executar uma função SUMIF no Pandas
Como realizar uma soma GroupBy no Pandas
Como somar colunas com base em uma condição no Pandas