python-pandas的基本用法04

pandas的基本用法04-drop()

# -*- coding: utf-8 -*-
import numpy as np
from pandas import Series, DataFrame
from numpy.core.defchararray import index
from matplotlib.pyplot import axis

s = Series(np.arange(5.), index=[1,2,3,4,5])
newS = s.drop([1,3])
print newS
# 2    1
# 4    3
# 5    4
# dtype: float64

print 'DataFrame删除元素，可指定索引或列。'
df = DataFrame(np.arange(16).reshape(4,4),
index = ['a','b','c','d'],
columns=[1,2,3,4]
)

print df
#     1   2   3   4
# a   0   1   2   3
# b   4   5   6   7
# c   8   9  10  11
# b  12  13  14  15
print df.drop(['a', 'b']) #闭区间 ['a','b']
#     1   2   3   4
# c   8   9  10  11
# d  12  13  14  15
print df.drop([1,2], axis=1)
#     3   4
# a   2   3
# b   6   7
# c  10  11
# d  14  15