python - Unable to drop a column from pandas dataframe

Question

This question already has answers here: question from:https://stackoverflow.com/questions/65916422/how-to-solve-the-problem-keyerror-label-not-found-in-axis

Answer 1

You have to use the inplace and axis parameter:

parts_median_temp.drop('Flag_median', axis=1, inplace=True)

The default value of 'inplace' is False, and axis' default is 0. axis=0 means dropping by index, whereas axis=1 will drop by column.