在Python的pandas库中,drop函数用于删除DataFrame或Series中的行或列。
X = df.drop(columns=['小明', '小红'])
这行代码的意思是,从df这个DataFrame中删除名为’小明’和’小红’的列,并将结果赋值给X。
详细解释
:
首先,让我们创建一个简单的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