利用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():数据类型转换

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

相关推荐
Deng94520131418 小时前
基于Python的旅游数据可视化应用
python·numpy·pandas·旅游·数据可视化技术
liuweidong08021 天前
【Pandas】pandas DataFrame from_records
pandas
大数据魔法师2 天前
基于Pandas和FineBI的昆明职位数据分析与可视化实现(五) - 基于随机森林算法预测职位分类
算法·pandas
Chasing__Dreams3 天前
python--杂识--18.1--pandas数据插入sqlite并进行查询
python·sqlite·pandas
超龄超能程序猿4 天前
(3)机器学习小白入门 YOLOv: 解锁图片分类新技能
python·numpy·pandas·scipy
好开心啊没烦恼5 天前
Python 数据分析:计算,分组统计1,df.groupby()。听故事学知识点怎么这么容易?
开发语言·python·数据挖掘·数据分析·pandas
大虫小呓6 天前
Python 处理 Excel 数据 pandas 和 openpyxl 哪家强?
python·pandas
好开心啊没烦恼7 天前
Python 数据分析:numpy,抽提,整数数组索引与基本索引扩展(元组传参)。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy·pandas
旷世奇才李先生8 天前
Pandas 安装使用教程
pandas