pandas extract specific columns code example

Example 1: pandas iloc select certain columns

dataframe.iloc[:,[1,2]]

Example 2: df only take 2 columns

df1 = df[['a', 'b']]