Let’s start with a data frame that has two columns:
import pandas as pd
df = pd.DataFrame({'A' : [1,2,4], 'B' : [2,2,2]})