pandas中drop使用

pandas中的drop()函数可以用于删除DataFrame或Series中的行或列。它有两个主要参数：labels和axis。labels参数指定要删除的行或列的名称，axis参数指定要删除的轴（0表示行，1表示列）。默认情况下，drop()函数不会更改原始数据，而是返回一个新的DataFrame或Series。如果要在原始数据上进行更改，则可以将inplace参数设置为True。

以下是一些使用drop()函数的示例：

1. 删除DataFrame中的一列：

import pandas as pd
import numpy as np
from pandas import DataFrame

test = DataFrame(np.arange(16).reshape((4,4)),index=['0','1','2','3'],columns=['a','b','c','d'])
test1 = test.drop(['a','b','c'],axis=1) # 删除a,b,c列
print(test1)

输出：

    d
0   3
1   7
2  11
3  15

2. 删除DataFrame中的一行：

test2 = test.drop(['1']) # 删除1那一列，axis默认为0
print(test2)

输出：

    a   b   c   d
0   0   1   2   3
2   8   9  10  11
3  12  13  14  15

3. 删除Series中的一个元素：

s = pd.Series([1,2,3,4,5], index=['a','b','c','d','e'])
s1 = s.drop(['c'])
print(s1)

输出：

a    1
b    2
d    4
e    5
dtype: int64