pandas中drop用法

pandas中的drop()函数可以用于删除DataFrame中的行或列。下面是一些使用drop()函数的例子：

1. 删除DataFrame中的一行

import pandas as pd
import numpy as np

df = pd.DataFrame(np.arange(12).reshape(3,4), columns=['A', 'B', 'C', 'D'])
print("原始DataFrame：\n", df)

df = df.drop(1) # 删除第2行
print("删除第2行后的DataFrame：\n", df)

2. 删除DataFrame中的多行

import pandas as pd
import numpy as np

df = pd.DataFrame(np.arange(12).reshape(3,4), columns=['A', 'B', 'C', 'D'])
print("原始DataFrame：\n", df)

df = df.drop([0, 2]) # 删除第1行和第3行
print("删除第1行和第3行后的DataFrame：\n", df)

3. 删除DataFrame中的一列

import pandas as pd
import numpy as np

df = pd.DataFrame(np.arange(12).reshape(3,4), columns=['A', 'B', 'C', 'D'])
print("原始DataFrame：\n", df)

df = df.drop('B', axis=1) # 删除'B'列
print("删除'B'列后的DataFrame：\n", df)

4. 删除DataFrame中的多列

import pandas as pd
import numpy as np

df = pd.DataFrame(np.arange(12).reshape(3,4), columns=['A', 'B', 'C', 'D'])
print("原始DataFrame：\n", df)

df = df.drop(['B', 'D'], axis=1) # 删除'B'列和'D'列
print("删除'B'列和'D'列后的DataFrame：\n", df)