subtraction between column a and column b in python in python

main.py
import pandas as pd

# create sample dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# perform subtraction of column B from column A
df['C'] = df['A'] - df['B']

# view the resulting dataframe
print(df)

# output:
#    A  B  C
# 0  1  4 -3
# 1  2  5 -3
# 2  3  6 -3

# if you want to perform the operation on a specific subset of columns, you can specify them as a list
df['C'] = df[['A', 'B']].apply(lambda x: x[0] - x[1], axis=1)
447 chars
20 lines