Python | 熊猫 Dataframe.pop()

简介

在 Pandas 中，DataFrame 是最基本的数据结构之一，它是用于处理表格类数据的二维数组。

DataFrame.pop() 方法用于从 DataFrame 中获取并移除指定列。

语法

DataFrame.pop(item)

参数：

• item：要获取的列的名称或索引（整数或布尔值 Series）或列（位置）。

返回值： 返回指定列项，以及从 DataFrame 中移除该列。

示例

假设现在有以下 DataFrame：

import pandas as pd

data = {
    'Name': ['Alice', 'Bob', 'Charles', 'David', 'Eric'],
    'Age': [25, 32, 18, 47, 22],
    'Gender': ['F', 'M', 'M', 'M', 'M'],
    'City': ['New York', 'Paris', 'London', 'Berlin', 'Sydney'],
}

df = pd.DataFrame(data)

现在要获取并移除 'Age' 列：

age = df.pop('Age')

此时，DataFrame df 中已经没有 'Age' 列了，而 age 中则包含了 'Age' 列的数据。

注意事项

• 当使用 DataFrame.pop() 方法时，原 DataFrame 中的数据会受到影响，推荐在原 DataFrame 上操作时使用此方法。如果不想在原数据上进行操作可以使用 .copy() 方法先复制一份。
• 如果尝试获取不存在的列，会触发 KeyError 异常。
• DataFrame.pop() 方法只能获取一列数据，并且返回值是一个 Series 对象。如果想要获取多列数据可以使用 DataFrame.loc/iloc 等方法。

以上就是 Python | 熊猫 Dataframe.pop() 的介绍，希望对你有所帮助。