要删除DataFrame中的列,可以使用drop()
函数。
示例代码如下:
import pandas as pd
# 创建DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 删除列B
df = df.drop('B', axis=1)
print(df)
输出结果:
A C
0 1 7
1 2 8
2 3 9
在drop()
函数中,axis=1
表示按列删除,axis=0
表示按行删除。