Adicionando colunas a um dataframe do pandas

De certa forma, os dataframes agem de maneira muito semelhante aos dicionários Python, pois você adiciona facilmente novas colunas. É tão simples como:

df = pandas.DataFrame.from_csv('my_data.csv')  # fake data
df
['diff_A_B'] = df['A'] - df['B']

Você também pode usar o assignmétodo para retornar uma cópia modificada

df2 = df.assign(diff_col=df['A'] - df['B'])