如何用Python的Pandas为表格指定末行添加自定义标签(含示例)
使用Pandas实现行标记的两种场景
以下是针对两个需求的具体实现代码,基于你提供的示例数据展开:
示例输入数据构造
首先构造和示例一致的输入表格:
import pandas as pd data = {"sample": ["abc", "bcd", "cde", "edc"]} df = pd.DataFrame(data)
需求2:最后两行标记"test",其余行标记"-"
先给所有行默认赋值"-",再通过索引定位最后两行覆盖值:
# 添加result列并默认赋值为"-" df["result"] = "-" # 定位最后两行,将result列设为"test" df.loc[df.index[-2:], "result"] = "test"
执行后得到的结果和示例输出一致:
sample result 0 abc - 1 bcd - 2 cde test 3 edc test
需求1:最后三行标记"a"
逻辑和需求2一致,仅调整定位的行范围:
# 添加result列并默认赋值为"-" df["result"] = "-" # 定位最后三行,将result列设为"a" df.loc[df.index[-3:], "result"] = "a"
执行后结果如下:
sample result 0 abc - 1 bcd a 2 cde a 3 edc a
内容的提问来源于stack exchange,提问作者Lintang Gilang Pratama
相关产品推荐
相关产品推荐

