pandas中的drop

在pandas中，drop()函数用于删除DataFrame中的行或列。下面是两个例子来演示pandas中的drop()函数的用法：

1. 删除DataFrame中的行：

import pandas as pd
import numpy as np

df = pd.DataFrame({'name': ['张丽华', '李诗诗', '王语嫣', '赵飞燕', '阮玲玉'],
                   'sex': ['girl', 'woman', np.nan, 'girl', 'woman'],
                   'age': [22, np.nan, 16, np.nan, 27]})

print(df)  # 输出原始DataFrame

print("--drop_index--")
# 使用drop()函数删除指定的行
df = df.drop(index=[0, 1, 2])
print(df)  # 输出删除指定行后的DataFrame

2. 删除DataFrame中的列：

import pandas as pd
import numpy as np

df1 = pd.DataFrame(np.arange(36).reshape(6, 6), columns=list('ABCDEF'))

print(df1)  # 输出原始DataFrame

print("--drop_columns--")
# 使用drop()函数删除指定的列
df1 = df1.drop(columns=['B', 'D'])
print(df1)  # 输出删除指定列后的DataFrame