1. 程式人生 > >pandas學習筆記-丟棄指定軸上的項

pandas學習筆記-丟棄指定軸上的項

丟棄某條軸上的一個或多個項很簡單,只要有一個索引陣列或列表即可。由於需要執行一些資料整理和集合邏輯,所以drop方法返回的是一個在指定軸上刪除了指定值的新物件:

obj = pd.Series(np.arange(5.),index=['a','b','c','d','e'])
new_obj = obj.drop('c')
print new_obj
結果
a    0.0
b    1.0
d    3.0
e    4.0
dtype: float64
print obj.drop(['d','c'])
結果
a    0.0
b    1.0
e    4.0
dtype: float64

對於DataFrame,可以刪除任意軸上的索引值:

data = pd.DataFrame(np.arange(16).reshape(4,4),
                    index=['Ohio','Colorado','Utah','New York'],
                    columns=['one','two','three','four'])
print data
print data.drop(['Colorado','Ohio'])
結果
          one  two  three  four
Ohio        0    1
2 3 Colorado 4 5 6 7 Utah 8 9 10 11 New York 12 13 14 15 one two three four Utah 8 9 10 11 New York 12 13 14 15
print data.drop('two',axis=1)
結果
          one  three  four
Ohio        0      2     3
Colorado    4
6 7 Utah 8 10 11 New York 12 14 15
print data.drop(['two','four'],axis=1)
結果
          one  three
Ohio        0      2
Colorado    4      6
Utah        8     10
New York   12     14