pandas习题 104：按工作年限重复数据

看过来

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

（编码题）人事部在做“节日福利发放表”。

规则：

请根据“已工作年数”把每位员工复制成对应行数，以便后续直接按行打印礼品券。

原始数据：

import pandas as pd

staff = pd.DataFrame(
    {'姓名': ['Alice', 'Bob', 'Cindy'],
     '已工作年数': [2, 1, 3]}
)

用 pandas .repeat() 一行代码完成礼品券行数展开，并打印结果。

参考答案

代码如下：

staff.loc[staff.index.repeat(staff['已工作年数'])]

输出：

      姓名  已工作年数
0  Alice      2
0  Alice      2
1    Bob      1
2  Cindy      3
2  Cindy      3
2  Cindy      3

（完）