在Python的pandas库中,drop函数用于删除DataFrame或Series中的行或列。
1. 例子说明
X = df.drop(columns=['小明', '小红'])
这行代码的意思是,从df这个DataFrame中删除名为’小明’和’小红’的列,并将结果赋值给X。
详细解释
:
- df 是一个DataFrame对象。
- drop 是DataFrame的一个方法,用于删除行或列。
- columns=[‘小明’, ‘小红’] 指定了要删除的列名。在这个例子中,我们删除了’小明’和’小红’这两列。
- X 是一个新的DataFrame,它包含了从df中删除’小明’和’小红’这两列后的结果。
注意:如果DataFrame中存在名为’小明’和’小红’的列,这行代码会成功执行。如果不存在这些列,则会引发一个KeyError。
2. 例子2详细说明:
首先,让我们创建一个简单的DataFrame:
import pandas as pd
data = {
'A': [1, 2, 3],
'B': [4, 5, 6],
'小明': ['a', 'b', 'c'],
'C': [7, 8, 9],
'小红': ['d', 'e', 'f']
}
df = pd.DataFrame(data)
print("原始DataFrame:")
print(df)
输出:
原始DataFrame:
A B 小明 C 小红
1 4 a 7 d
2 5 b 8 e
3 6 c 9 f
现在,我们使用drop函数删除名为'小明'和'小红'的列
:
X = df.drop(columns=['小明', '小红'])
print("删除列后的DataFrame:")
print(X)
输出:
删除列后的DataFrame:
A B C
1 4 7
2 5 8
3 6 9