how to drop a column permanently in pandas 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: pandas read_csv drop column
# Read column names from file
cols = list(pd.read_csv("sample_data.csv", nrows =1))
print(cols)
# Define unused cols
unused = ['col1', 'col2']
# Use list comprehension to remove the unwanted column in **usecol**
df= pd.read_csv("sample_data.csv", usecols =[i for i in cols if i not in unused])