pandas pop() 删除并返回指定列

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gr99123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

pandas 的 pop 方法用于从 DataFrame 中删除并返回指定列（或行，如果操作在行上）。

语法和参数说明

Pandas 中的 pop 方法用于移除并返回指定列（或行）的数据。它适用于 DataFrame 和 Series 对象。

DataFrame 的语法：

DataFrame.pop(item)

item：要移除并返回的列或行的标签。可以是列名（字符串）或行索引（整数）。

Series 的语法：

Series.pop(item)

item：要移除并返回的元素的索引标签。

返回值

对于 DataFrame：
- 返回被移除的列作为 Series 对象。
- 如果 item 是列名，则返回一个 Series，列名为 item，并从原始 DataFrame 中移除该列。
- 如果 item 是整数，则返回相应行作为 Series，并从原始 DataFrame 中移除该行。
对于 Series：
- 返回被移除的元素的值，并从原始 Series 中移除该元素。

使用场景

数据提取和转移：将某列的数据从 DataFrame 中移除，并将其存储在另一个变量中。
动态数据操作：根据特定条件动态移除和处理数据。
实时数据处理：在数据流处理中，用于动态更新和管理数据。

示例说明

示例 1：从 DataFrame 中移除并返回指定列

import pandas as pd

# 创建一个 DataFrame
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}
df = pd.DataFrame(data)

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

print("移除的列 'B':")
print(column_b)
print("\n剩余的 DataFrame:")
print(df)

输出：

移除的列 'B':
0    4
1    5
2    6
Name: B, dtype: int64

剩余的 DataFrame:
   A  C
0  1  7
1  2  8
2  3  9

示例 2：从 Series 中移除并返回指定元素

import pandas as pd

# 创建一个 Series
s = pd.Series([10, 20, 30, 40], index=['a', 'b', 'c', 'd'])

# 使用 pop 方法移除并返回索引 'c' 处的元素
element_c = s.pop('c')

print("移除的元素值:", element_c)
print("\n剩余的 Series:")
print(s)

输出：

移除的元素值: 30

剩余的 Series:
a    10
b    20
d    40
dtype: int64

这些示例展示了 pop 方法在 DataFrame 和 Series 中的基本用法和效果，适用于从数据结构中动态移除并获取数据的场景。

参考

https://pandas.pydata.org/docs/reference/api/pandas.Series.pop.html
https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.pop.html

pandas pop() 删除并返回指定列

语法和参数说明

返回值

使用场景

示例说明

参考

相关内容