add axis = 1 to drop
By default, the drop() function uses axis=0, which refers to the rows of the DataFrame.To tell pandas to look at the columns instead, we must specify axis=1.
By default, the drop() function uses axis=0, which refers to the rows of the DataFrame.To tell pandas to look at the columns instead, we must specify axis=1.