Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

remove columns from a dataframe python

# Import pandas package
import pandas as pd

# create a dictionary with five fields each
data = {
'A':['A1', 'A2', 'A3', 'A4', 'A5'],
'B':['B1', 'B2', 'B3', 'B4', 'B5'],
'C':['C1', 'C2', 'C3', 'C4', 'C5'],
'D':['D1', 'D2', 'D3', 'D4', 'D5'],
'E':['E1', 'E2', 'E3', 'E4', 'E5'] }

# Convert the dictionary into DataFrame
df = pd.DataFrame(data)
#drop the 'A' column from your dataframe df 
df.drop(['A'],axis=1,inplace=True)
df

#-->df contains 'B','C','D' and 'E'
#in this example you will change your dataframe , if you don't want to ,
#just remove the in place parameter and assign your result to an other variable 

df1=df.drop(['B'],axis=1)
#-->df1 contains 'C','D','E'
df1
Source by app.dataquest.io #
 
PREVIOUS NEXT
Tagged: #remove #columns #dataframe #python
ADD COMMENT
Topic
Name
5+5 =