how to delete columns in dataframe python code example

Example 1: drop a column from dataframe

#To delete the column without having to reassign df
df.drop('column_name', axis=1, inplace=True)

Example 2: python code to drop columns from dataframe

# Let df be a dataframe
# Let new_df be a dataframe after dropping a column

new_df = df.drop(labels='column_name', axis=1)

# Or if you don't want to change the name of the dataframe
df = df.drop(labels='column_name', axis=1)

# Or to remove several columns
df = df.drop(['list_of_column_names'], axis=1)

# axis=0 for 'rows' and axis=1 for columns

Example 3: removing features pandas

df.drop([‘feature_1’,’feature_2’], inplace = True, axis = 1)

Example 4: how to drop a column by name in pandas

>>> midx = pd.MultiIndex(levels=[['lama', 'cow', 'falcon'],
...                              ['speed', 'weight', 'length']],
...                      codes=[[0, 0, 0, 1, 1, 1, 2, 2, 2],
...                             [0, 1, 2, 0, 1, 2, 0, 1, 2]])
>>> df = pd.DataFrame(index=midx, columns=['big', 'small'],
...                   data=[[45, 30], [200, 100], [1.5, 1], [30, 20],
...                         [250, 150], [1.5, 0.8], [320, 250],
...                         [1, 0.8], [0.3, 0.2]])
>>> df
                big     small
lama    speed   45.0    30.0
        weight  200.0   100.0
        length  1.5     1.0
cow     speed   30.0    20.0
        weight  250.0   150.0
        length  1.5     0.8
falcon  speed   320.0   250.0
        weight  1.0     0.8
        length  0.3     0.2