1. 程式人生 > >pandas的DataFrame、Series刪除列

pandas的DataFrame、Series刪除列

Series方法與DataFrame差不多,這裡只介紹後者如何使用,前者相似。

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

In [4]: df
Out[4]:
   A  B   C   D
0  0  1   2   3
1  4  5   6   7
2  8  9  10  11

axis=1(按列方向操作)、inplace=True(修改完資料,在原資料上儲存)

一:按標籤來刪除列

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

   A   D
0  0   3
1  4   7
2  8  11

二:按序號來刪除列

x = [1,2]    #刪除多列需給定列表,否則引數過多

df.drop(df.columns[x],axis=1,inplace=True)

   A   D
0  0   3
1  4   7
2  8  11

三:按序號來刪除行

df.drop([0,1],inplace=True)        #預設axis=0

   A  B   C   D
2  8  9  10  11