drop columns pandas code example
Example 1: python pandas drop
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
Drop columns
>>> df.drop(['B', 'C'], axis=1)
A D
0 0 3
1 4 7
2 8 11
>>> df.drop(columns=['B', 'C'])
A D
0 0 3
1 4 7
2 8 11
Example 2: delete columns pandas
df = df.drop(df.columns[[0, 1, 3]], axis=1) # df.columns is zero-based pd.Index
df.drop(['column_nameA', 'column_nameB'], axis=1, inplace=True)