利用Pandas进行条件替换与向前填充

目录

一、需求

示例数据:

python 复制代码
   项  目
0  1  0
1  0  1
2  0  2
3  0  3
4  0  4

使用 Pandas 实现当 目 列的值不为 0 时,将当前行的 项 列值修改为上一行的 项 列值:

python 复制代码
   项  目
0  1  0
1  1  1
2  1  2
3  1  3
4  1  4

二、实现代码

案例代码

python 复制代码
import pandas as pd

# 创建示例数据
df = pd.DataFrame({
    '项': [1, 0, 0, 0, 0],
    '目': [0, 1, 2, 3, 4]
})
print(df)

# 当`目`列不为0时,`项`列的值被替换为NaN,然后向前填充
df['项'] = df['项'].mask(df['目'] != 0).ffill().astype(int)

print(df)

详细解释

1. 导入库和创建数据

python 复制代码
import pandas as pd

# 创建示例数据
df = pd.DataFrame({
    '项': [1, 0, 0, 0, 0],
    '目': [0, 1, 2, 3, 4]
})
  • 首先导入Pandas库并简称为pd
  • 创建一个DataFrame df,包含两列:
    • "项"列:初始值为[1, 0, 0, 0, 0]
    • "目"列:初始值为[0, 1, 2, 3, 4]

初始数据如下:

复制代码
   项  目
0  1  0
1  0  1
2  0  2
3  0  3
4  0  4

2. 条件替换与填充

python 复制代码
df['项'] = df['项'].mask(df['目'] != 0).ffill().astype(int)

这行代码执行了多个操作,我们将其分解:

a. 条件掩码 - mask()

df['项'].mask(df['目'] != 0)

  • df['目'] != 0 创建一个布尔Series:[False, True, True, True, True]
  • mask()函数会用NaN替换满足条件(True)的位置
  • 结果:"项"列变为 [1, NaN, NaN, NaN, NaN]
b. 向前填充 - ffill()

.ffill()

  • 向前填充(Forward Fill)方法会用前一个有效值填充NaN
  • 对于我们的数据:
    • 第一个值是1(保持不变)
    • 后续NaN被前一个有效值1填充
  • 结果:"项"列变为 [1, 1, 1, 1, 1]
c. 类型转换 - astype(int)

.astype(int)

  • 将填充后的浮点数(因为NaN是浮点类型)转换为整数
  • 最终"项"列变为 [1, 1, 1, 1, 1]

3. 打印结果

python 复制代码
print(df)

输出结果:

复制代码
   项  目
0  1  0
1  1  1
2  1  2
3  1  3
4  1  4

三、实际应用场景

这种操作在实际数据处理中很有用,例如:

  1. 数据清洗:当某列(如"目")有特定值时,需要重置另一列(如"项")的值
  2. 时间序列处理:在某个事件发生后(如"目"列不为0表示事件发生),需要保持某个状态(如"项"列的值)
  3. 标记传播:将某个标记从特定点向前传播

四、可能的变体

  1. 向后填充 :使用.bfill()而不是.ffill()
  2. 不同条件:可以修改mask的条件逻辑
  3. 填充特定值 :可以使用.fillna()填充特定值而非前向填充

五、总结

这段代码展示了Pandas中几个强大功能的组合使用:

  • mask():基于条件替换值
  • ffill():向前填充缺失值
  • astype():数据类型转换

通过这种组合,可以高效地实现复杂的数据转换逻辑,这在数据预处理和分析中非常常见。理解这些基本操作的组合方式,可以帮助我们更灵活地处理各种数据清洗和转换任务。

相关推荐
qq_381454994 天前
Python Pandas完全指南:从核心数据结构到实战操作
pandas
TUTO_TUTO5 天前
【python-词汇指标提取工具开发】自学笔记(1)-后端程序
人工智能·python·pandas·visual studio code
葱明撅腚5 天前
shapely空间数据分析
python·pandas·gis·shapely
忘忧记6 天前
pandas基础三
pandas
叫我:松哥8 天前
基于scrapy的网易云音乐数据采集与分析设计实现
python·信息可视化·数据分析·beautifulsoup·numpy·pandas
测试摆渡媛8 天前
Excel模板填充工具(工具&脚本分享)
python·数据挖掘·pandas
_Soy_Milk8 天前
【算法工程师】—— Python 数据分析
python·数据分析·numpy·pandas·matplotlib
Data-Miner9 天前
类似Pandas AI的几个数据分析处理智能体介绍
人工智能·数据分析·pandas
智航GIS11 天前
11.18 自定义Pandas扩展开发指南:打造你的专属数据分析武器库
python·数据分析·pandas
人工干智能12 天前
你知道 Pandas 中 `pd.get_dummies()` 会生成哪些独热的新列么?
大数据·pandas