Run
df[['A', 'B']]
Output