data.table drop columns code example

Example 1: 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 2: drop columns

>>> df.drop(index='cow', columns='small')
                big
lama    speed   45.0
        weight  200.0
        length  1.5
falcon  speed   320.0
        weight  1.0
        length  0.3