how to delete a column in python dataframe 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: how to drop columns in pandas

>>>df = pd.DataFrame(np.arange(12).reshape(3, 4), 
                     columns=['A', 'B', 'C', 'D'])
>>>df
   A  B   C   D
0  0  1   2   3
1  4  5   6   7
2  8  9  10  11

>>> df.drop(['B', 'C'], axis=1)
   A   D
0  0   3
1  4   7
2  8  11

OR

>>> df.drop(columns=['B', 'C'])
   A   D
0  0   3
1  4   7
2  8  11

Example 3: dataframe delete row

df.drop(df.index[2])

Example 4: drop a column from dataframe

#to drop by column number instead of by column label, try this to delete, e.g. the 1st, 2nd and 4th columns
df = df.drop(df.columns[[0, 1, 3]], axis=1)  # df.columns is zero-based pd.Index