dataframe extract df column code example

Example 1: python - subset specific columns name in a dataframe

columns = ['b', 'c']
df1 = pd.DataFrame(df, columns=columns)

Example 2: df only take 2 columns

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