要刪除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
表示按行刪除。