DataFrame-删除行列

1 构造数据

import pandas as pd
import numpy as np
df = pd.DataFrame(np.arange(12).reshape(3,4), columns=['A', 'B', 'C', 'D'])
df

2 删除列, 两种方法等价

df.drop(['B', 'C'], axis=1)

df.drop(columns=['B', 'C'])

第一种方法下删除column一定要指定axis=1,否则会报错

df.drop(['B', 'C'])

KeyError: “[‘B’ ‘C’] not found in axis”

3 删除行，两种方式等价

df.drop([0, 1])

df.drop(index = [0, 1])