如何在pandas DataFrame中筛选薪资较上月增长的月份?
筛选薪资较上月增长的月份
首先需要处理薪资列的格式——当前Salary列是带$的字符串,无法直接进行数值比较:
# 去除$符号并转换为整数类型 m_salary_df['Salary'] = m_salary_df['Salary'].str.replace('$', '', regex=False).astype(int)
接下来利用pandas的shift()方法获取上月薪资数据,筛选出当前薪资大于上月薪资的行:
# 一步筛选出薪资增长的月份 growth_months = m_salary_df[m_salary_df['Salary'] > m_salary_df['Salary'].shift(1)]['Month']
执行后输出结果:
print(growth_months.tolist()) # 输出: ['Feb.']
这种基于pandas向量化操作的方式,比循环更高效简洁,也能避免循环逻辑容易出错的问题。
内容的提问来源于stack exchange,提问作者cuddlycritter22
相关产品推荐
相关产品推荐

