Como subtrair duas colunas no pandas dataframe

Você pode usar a seguinte sintaxe para subtrair uma coluna de outra em um DataFrame do pandas:

 #subtract column 'B' from column 'A'
df[' AB '] = df. A - df. B

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Subtraia duas colunas no Pandas

O código a seguir mostra como subtrair uma coluna de outra em um DataFrame do pandas e atribuir o resultado a uma nova coluna:

 import pandas as pd

#createDataFrame 
df = pd. DataFrame ({' A ': [25, 12, 15, 14, 19, 23, 25, 29],
                   ' B ': [5, 7, 8, 9, 12, 9, 12, 4],
                   ' C ': [11, 8, 10, 6, 6, 5, 9, 12]})

#subtract column B from column A
df[' AB '] = df. A - df. B

#view DataFrame
df

        A B C AB
0 25 5 11 20
1 12 7 8 5
2 15 8 10 7
3 14 9 6 5
4 19 12 6 7
5 23 9 5 14
6 25 12 9 13
7 29 4 12 25

A nova coluna chamada ‘ AB ’ exibe os resultados da subtração dos valores da coluna B dos valores da coluna A.

Exemplo 2: subtraia duas colunas com valores ausentes

Se subtrairmos uma coluna da outra em um DataFrame do pandas e houver valores faltantes em uma das colunas, o resultado da subtração será sempre um valor faltante:

 import pandas as pd
import numpy as np

#create DataFrame with some missing values
df = pd. DataFrame ({' A ': [25, 12, 15, 14, 19, 23, 25, 29],
                   ' B ': [5, 7, np. no , 9, 12, np. no , 12, 4],
                   ' C ': [np. no , 8, 10, 6, 6, 5, 9, 12]}) 

#subtract column B from column A
df[' AB '] = df. A - df. B

#view DataFrame
df

	A B C AB
0 25 5.0 NaN 20.0
1 12 7.0 8.0 5.0
2 15 NaN 10.0 NaN
3 14 9.0 6.0 5.0
4 19 12.0 6.0 7.0
5 23 NaN 5.0 NaN
6 25 12.0 9.0 13.0
7 29 4.0 12.0 25.0

Se desejar, você pode substituir todos os valores ausentes no dataFrame por zeros usando a função df.fillna(0) antes de subtrair uma coluna da outra:

 import pandas as pd
import numpy as np

#create DataFrame with some missing values
df = pd. DataFrame ({' A ': [25, 12, 15, 14, 19, 23, 25, 29],
                   ' B ': [5, 7, np. no , 9, 12, np. no , 12, 4],
                   ' C ': [np. no , 8, 10, 6, 6, 5, 9, 12]}) 

#replace all missing values with zeros
df = df. fillna ( 0 )

#subtract column B from column A
df[' AB '] = df. A - df. B

#view DataFrame
df
	A B C AB
0 25 5.0 0.0 20.0
1 12 7.0 8.0 5.0
2 15 0.0 10.0 15.0
3 14 9.0 6.0 5.0
4 19 12.0 6.0 7.0
5 23 0.0 5.0 23.0
6 25 12.0 9.0 13.0
7 29 4.0 12.0 25.0

Recursos adicionais

Como adicionar linhas a um DataFrame do Pandas
Como adicionar um array Numpy a um DataFrame do Pandas
Como contar o número de linhas no Pandas DataFrame