将收入段转换成收入

最新推荐文章于 2024-08-20 13:04:52 发布

Sheyueyu

最新推荐文章于 2024-08-20 13:04:52 发布

阅读量111

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/sheyueyu/article/details/127836136

版权

python 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这段代码定义了一个函数`cal_income`，根据给定的收入段和家庭人口数，从`income_map`字典中随机选取一个收入值并计算家庭总收入。然后使用pandas的apply方法将该函数应用于`choosers`数据框的`income_segment`和`household_pop_num`列，生成新的`imcome`列表示家庭总收入。

摘要由CSDN通过智能技术生成

#将收入段转换成收入
imcome_map = {
    1:[2000,3000],
    2:[3000,3500],
    3:[3500,6000],
    4:[6000,8000],
    5:[8000,10000],
    6:[10000,15000]
}
def cal_income(income_segment,household_pop_num):
    per_income  = random.choice(imcome_map[income_segment])   #在对应的收入区间里面进行选择
    h_income_all  = per_income*household_pop_num
    return h_income_all


choosers['imcome'] = choosers[['income_segment','household_pop_num']].apply(lambda x:cal_income(x['income_segment'],x['household_pop_num']),axis =1)
choosers