Python pandas.DataFrame.pop函数方法的使用

DataFrame.pop(self: ~FrameOrSeries, item) → ~FrameOrSeries       [source]

返回item并从frame中删除。如果找不到，会引发KeyError。

例子

import pandas as pd

# 创建一个示例 DataFrame
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}

df = pd.DataFrame(data)

# 输出原始 DataFrame
print("原始 DataFrame:")
print(df)

# 使用 pop 方法删除并返回列 'B'
b_column = df.pop('B')

# 输出删除后的 DataFrame
print("\n删除列 'B' 后的 DataFrame:")
print(df)

# 输出被删除的列
print("\n被删除的列 'B':")
print(b_column)